2016-11-28 2 views
0

Я определил внешнюю функцию подсчета очков, чтобы определить точность кластеров для рукописных цифр, но не могу понять, как использовать эту функцию в сочетании с GridSearchCV.Можно ли использовать Grid Search с внешней функцией подсчета очков?

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))  
+0

Что такое y_pred_km? –

+0

Предполагается, что это метки кластеров, полученные из kmeans. –

ответ

0

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) 
+0

Большое вам спасибо! –

Смежные вопросы