GridSearchCV в лесу изоляции, скажем, цель мультиклассовая, хотя она двоичная - PullRequest
0 голосов
/ 20 марта 2019

Я пытаюсь оптимизировать параметры моей модели Isolation Forest с помощью функции GridSearchCv. У меня есть набор данных, касающихся обнаружения мошенничества, где целевая переменная имеет значения [0,1].

Когда я пытаюсь использовать точность в качестве секретаря и запускаю функцию:

clf = IsolationForest(n_jobs=-1)
param_grid = {'n_estimators' : range(20,230,30)}

scorers = {'precision_score': make_scorer(precision_score)}

skf = StratifiedKFold(n_splits=10)
grid_search = GridSearchCV(clf, param_grid, scoring=scorers, refit=refit_score,
                       cv=skf, return_train_score=True, n_jobs=-1)

grid_search.fit(X_train.values, y_train.values)

Я всегда получаю сообщение об ошибке:

Target is multiclass but average='binary'. Please choose another average setting.

Я не понимаю, почему, я пробовал другой набор данных с двоичной целевой переменной, и это сработало.

...