Catboost预测概率为负

Question

我是catboost的新手，我正尝试在R中的虹膜数据集上进行单次迭代。我从预测函数的输出中注意到负的概率。你知道为什么会这样吗？另外，有没有一种方法可以在R中绘制catboost树？

test <- as.data.frame(iris)
test$y <- ifelse(test$Species=="setosa",1,0)
x_iris <- test[,c("Sepal.Length","Sepal.Width","Petal.Length","Petal.Width")]
y_iris <- test[,"y"]

train <- as.data.frame(lapply(x_iris, as.numeric))
train_pool <- catboost.load_pool(data = train, label = y_iris)

model <- catboost.train(train_pool,  NULL,
                        params = list(loss_function = 'Logloss',
                                      iterations = 1, metric_period=10))
print(prediction)

Answer 1

在python中:我们必须设置prediction_type='Probability'