Можно ли использовать 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)
Другие вопросы по тегам