是否可以使用網格搜索與外部定義的評分函數？

我定義了一個外部評分函數來確定手寫數字的聚類準確性，但無法弄清楚如何將該函數與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))

來源

2016-11-28 Abdullah Nazir

什麼是y_pred_km？ –

假設是從kmeans獲得的集羣標籤。 –

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)

來源

2016-11-28 17:40:20

非常感謝！ –

是否可以使用網格搜索與外部定義的評分函數？

回答

相關問題