随机森林模型sklearn_sklearn随机森林调参小结

param_test2 = {'max_depth': range(3, 14, 2), 'min_samples_split': range(50, 201, 20)}

gsearch2 = GridSearchCV(

estimator=RandomForestClassifier(n_estimators=60, min_samples_leaf=20, max_features='sqrt',

oob_score=True, random_state=10),

param_grid=param_test2, scoring='roc_auc', iid=False, cv=5)

gsearch2.fit(X, y)

print(gsearch2.grid_scores_, gsearch2.best_params_, gsearch2.best_score_)

"""

[mean: 0.79379, std: 0.02347, params: {'max_depth': 3, 'min_samples_split': 50},

mean: 0.79339, std: 0.02410, params: {'max_depth': 3, 'min_samples_split': 70},

mean: 0.79350, std: 0.02462, params: {'max_depth': 3, 'min_samples_split': 90},

mean: 0.79367, std: 0.02493, params: {'max_depth': 3, 'min_samples_split':