Имею только 2 класса, но почему я получаю ValueError: Цель является многоклассовой, но среднее значение = «двоичное». По ⇐ Python

[Anonymous]

Я пытаюсь построить модель прогнозирования оттока сотрудников с помощью GridSearchCV на OneClassSVM. Мой код выглядит следующим образом:

Код: Выделить всё

from sklearn.svm import OneClassSVM
from sklearn.model_selection import GridSearchCV
from sklearn.metrics import precision, recall, f1_score

grid = {
'nu' : [0.02, 0.03],
'kernel' : ['poly', 'sigmoid', 'linear', 'rbf'],
'degree' : [1,3,5,7],
'gamma' : [.01, .1, 1, 10, 500]
'coeff0' : [1,2,3]
}

mod = GridSearchCV(OneClassSVM(),
grid,
cv = 3,
scoring = ['f1', 'precision', 'recall'],
refit = 'f1',
return_train_score = True)

mod.fit(x_train, y_train)

Здесь у y есть только 0 и 1 (по сути, смирился или нет), но не знаю, почему я получаю:

Код: Выделить всё

ValueError: Target is multiclass but average='binary'. Please choose another average setting, one of [None, 'micro', 'macro', 'weighted']

Может кто-нибудь помочь мне с этим.

Подробнее здесь: https://stackoverflow.com/questions/793 ... ass-but-av