Можно ли использовать Grid Search с внешней функцией оценки?
Я определил внешнюю функцию оценки для определения точности кластера для рукописных цифр, но не могу понять, как я могу использовать эту функцию в сочетании с GridSearchCV.
s = GridSearchCV(estimator = pipe_cl,
param_grid = param_distributions,
scoring = my_scorer,
n_jobs=-1)
gs.fit(X_train, y_train)
y_pred_km = gs.predict(X_test)
my_scorer = make_scorer(clustering_accuracy_score(y_test, y_pred_km))
1 ответ
Решение
make_scorer
возвращает новую функцию, она принимает функцию в качестве аргумента. Так что это должно работать:
my_scorer = make_scorer(clustering_accuracy_score)
gs = GridSearchCV(estimator = pipe_cl,
param_grid = param_distributions,
scoring = my_scorer,
n_jobs=-1)
gs.fit(X_train, y_train)