Вероятности прогноза Catboost отрицательны - PullRequest
Новая
       86

Вероятности прогноза Catboost отрицательны

0 голосов
/ 13 июля 2020

Я новичок в catboost и пытаюсь запустить алгоритм для набора данных iris в R за одну итерацию. Я заметил отрицательные вероятности на выходе функции прогнозирования. Есть идеи, почему это происходит? Кроме того, есть ли способ построить дерево котов в R? 

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)
...