风一样 冯伊洋
关注数: 13 粉丝数: 28 发帖数: 573 关注贴吧数: 20
问个问题,这部分代码没看懂 test_results = pd.read_csv('test_results.csv') def modelfit(alg, dtrain, dtest, predictors,useTrainCV=True, cv_folds=5, early_stopping_rounds=50): '''<br data-filtered="filtered">   功能:训练,测试,输出AUC,画出重要特征的功能 参数:alg是分类器,dtrain是训练集(包括label),dtest是测试集(不包括label),predictors是要参与训练的特征(不包括label), useTrainCV是是否要交叉验证,cv_folds是交叉验证的折数,early_stopping_rounds是到指定次数就停止继续迭代<br data-filtered="filtered"> ''' if useTrainCV: xgb_param = alg.get_xgb_params() xgtrain = xgb.DMatrix(dtrain[predictors].values, label=dtrain[target].values) xgtest = xgb.DMatrix(dtest[predictors].values) cvresult = xgb.cv(xgb_param, xgtrain, num_boost_round=alg.get_params()['n_estimators'], nfold=cv_folds, metrics='auc', early_stopping_rounds=early_stopping_rounds, show_progress=False) alg.set_params(n_estimators=cvresult.shape[0]) #训练 alg.fit(dtrain[predictors], dtrain['Disbursed'],eval_metric='auc') #预测 dtrain_predictions = alg.predict(dtrain[predictors]) dtrain_predprob = alg.predict_proba(dtrain[predictors])[:,1] ———————————————— 版权声明:本文为CSDN博主「语亦情非」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。 原文链接:http://tieba.baidu.com/mo/q/checkurl?url=https%3A%2F%2Fblog.csdn.net%2Fa1272899331%2Farticle%2Fdetails%2F104912984&urlrefer=034aad1575e91eaf32e0b57ffbdcddb2 定义了个函数modelfit,但是我也没看见他在下面的代码调用这个函数,为什么捏
1 下一页