python_交叉网格搜索

python_交叉网格搜索

### 3.2 交叉网格搜索
#grid_search
parameters = {'max_depth':[3,4,5,6,7]} #定义网格搜索参数
RF_grid = grid_search.GridSearchCV(estimator=RandomForestClassifier(),#模型
                                   parameters = parameters,#参数
                                   error_score='roc_auc') #评价指标
RF_grid.fit(X=temp.iloc[0:100000,:],y = [int(x>20) for x in data.cust_age.iloc[0:100000]])
print(RF_grid.best_estimator_)
print(RF_grid.best_params_)
RF_grid.grid_scores_