2016-05-26 5 views
0

グリッド検索CVを実行した後、将来的に使用するには最高のモデルを選んでください。私はグリッド検索から最高のモデルをピックルする方法はありますか?

grid_search = GridSearchCV(SVC_clf, parameters, n_jobs=-1, verbose=1, cv = 3) 

gs = grid_search.fit(X, Y) 


with open('../MLA (pickled)/SVC.txt','w') as f: 
    s = pickle.dumps(gs) 
    f.write(s) 

、その後

with open('../MLA (pickled)/SVC.txt','r') as f: 
    clf = pickle.loads(f.read()) 
clf.fit(X_new,Y_new) 

ような何かをするとき、グリッド検索が再び開始されます。特に、パラメータグリッドが非常に大きい場合、これは非常に苦痛になります。

グリッド検索から「ベスト」モデルをどのようにピックルできますか?

答えて

0

わかりました。 gsbest_estimator_と思われます。だから、私はピックルしてくださいgs.best_estimator_

関連する問題