Tune-sklearn的小麻烦
该文是我在实际运用时产生的问题,可能不具有一般性。本人水平有限,请大家指教。
报错
在使用TuneGridSearchCV对RandomForestClassifier进行超参数调优,报错:
This TuneGridSearchCV instance is not fitted yet.
代码
param_grid={
'n_estimators':[100,200,300],
'max_depth':range(5,30,5),
'min_samples_split':[2,5,10,15],
'min_samples_leaf':[1,2,5,10,15],
'max_features':['log2','sqrt']
}
forest_clf=RandomForestClassifier(random_state=42)
grid_search=TuneGridSearchCV(forest_clf,param_grid,cv=5,scoring='accuracy',use_gpu=True)
grid_search.fit(X_train,y_train)
grid_search.score(X_test,y_test)
摸索出的方法
1.删除use_gpu参数,具体原因,或者因为配置问题,我还不了解
2.Tune默认n_jobs=-1,我的代码运行过程中出现内存溢出的情况,故我改成了总CPU核心数-2.