2016-05-26 3 views
0

Nachdem ich den CV für die Rastersuche durchgeführt habe, möchte ich das beste Modell auswählen, das in Zukunft verwendet werden soll. Wenn ich etwas tun wieWie kann ich das beste Modell aus einer Rastersuche einlegen?

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) 

und dann

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

die Rastersuche erneut gestartet. Dies wäre extrem schmerzhaft, besonders wenn das Parameterraster sehr groß wäre.

Wie kann ich das "beste" Modell aus der Rastersuche einlegen?

Antwort

0

Oh, ich verstehe. Es scheint gs als best_estimator_. Also sollte ich gs.best_estimator_ beizen.

Verwandte Themen