加入MySQL 连接

models/xgboost.py

@@ -80,7 +80,7 @@ def modelfit(clf, dftrain, features, resp,useTrainCV = True, kfold=10, eval_metr
     if useTrainCV:
         # kflod = StratifiedKFold(n_splits=kfold, shuffle=True, random_state=7)
         xgb_param = clf.get_xgb_params()
-        xgtrain = xgb.DMatrix(dftrain[features].values, label=dftrain[resp].values)
+        xgtrain = xgb.DMatrix(dftrain[features], label=dftrain[resp])
         cvresult = xgb.cv(xgb_param, xgtrain, num_boost_round=clf.get_params()['n_estimators'], nfold=kfold,
             metrics=eval_metric, early_stopping_rounds=early_stopping_rounds,verbose_eval=True)
         clf.set_params(n_estimators=cvresult.shape[0])