看到好的代码,就记录下来,方便以后使用
# 引入相关包
import pandas as pd
import lightgbm as lgb
from sklearn.model_selection import StratifiedKFold
# 假设这里准备好了训练数据train_data,它是一个pandas的dataframe,包括特征列和score列
train_label = train_data['score']
# 初始化一个k-fold生成器
NFOLDS = 5
kfold = StratifiedKFold(n_splits=NFOLDS, shuffle=True, random_state=2019)
kf = kfold.split(train_data, train_label)
cv_pred = np.zeros(test_data.shape[0])
valid_best_l2_all = 0
feature_importance_df = pd.DataFrame()
# 执行训练
for i, (train_fold