sklearn - 网格搜索的使用

GridSearchCv 来生成超参数的组合,自动选择效果最好的 参数。
api :

from sklearn.model_selection import GridSearchCV

例子:

rfc  = RandomForestClassifier()
para_dict ={
     'n_estimators':[88,99,100,111,122,120],'max_depth':[8,9,10,13,14]}
gcv = GridSearchCV(para_dict ,param_grid=para_dict ,cv = 2)
gcv.fit()
gcv.predict()
gcv.score()
print(gscv.best_params_)

gscv.best_params_ 查看到最佳参数

你可能感兴趣的:(sklearn,机器学习)