如何从网格搜索中挑选最佳模型?

时间:2016-05-26 05:17:04

标签: scikit-learn

在我执行网格搜索CV之后,我想挑选出将来使用的最佳模型。当我做

之类的事情
<servlet>
    <servlet-name>Jersey Web Application</servlet-name>
    <servlet-class>org.glassfish.jersey.servlet.ServletContainer</servlet-class>
    <init-param>
        <param-name>jersey.config.server.provider.packages</param-name>
        <param-value>com.two95.restful</param-value>
    </init-param>
    <load-on-startup>1</load-on-startup>
</servlet>

然后

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)

再次启动网格搜索。这将非常痛苦,特别是如果参数网格非常大。

如何从网格搜索中挑选出“最佳”模型?

1 个答案:

答案 0 :(得分:0)

哦,我明白了。看起来gsbest_estimator_。所以我应该挑选gs.best_estimator_