2016-11-28 90 views
0

我定义了一个外部评分函数来确定手写数字的聚类准确性,但无法弄清楚如何将该函数与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))  
+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

非常感谢! –

相关问题