2016-11-28 32 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

ありがとうございました! –

関連する問題