因为自己电脑安装xgboost始终有问题,所以只优化了四个模型。详细代码如下所示:
##LR法的优化过程
ltc_param = {'penalty':['l1', 'l2'], 'C':[0.0001, 0.001, 0.01, 0.1, 1.0]}
ltc_grid = GridSearchCV(ltc, ltc_param, cv = n_fold, scoring=scoring, n_jobs=1)
ltc_grid.fit(X_train, y_train)
print("LR最优分数为:", ltc_grid.best_score_) #最优分数
print("LR最优参数为:", ltc_grid.best_params_) #最优参数display(pd.DataFrame(ltc_grid.cv_results_).T)
##决策树的优化过程
dtc_param = {'max_depth':range(1,10)}
dtc_grid = GridSearchCV(dtc, dtc_param, cv=n_fold, scoring=scoring, n_jobs=-1)
dtc_grid.fit(X_train, y_train)
print("决策树最优分数为:", dtc_grid.best_score_)
print("决策树最优参数为:", dtc_grid.best_params_)
##SVM的优化过程
svc_param = {'C': [0.001, 0.01, 0.1]}
svc_grid = GridSearchCV(svc, svc_param, cv=n