from sklearn.model_selection import KFold
分离数据集,方便进行交叉验证
X_train = data.loc[data[‘sample’]‘train’, :].drop([‘id’,‘issueDate’,‘isDefault’, ‘sample’], axis=1)
X_test = data.loc[data[‘sample’]‘test’, :].drop([‘id’,‘issueDate’,‘isDefault’, ‘sample’], axis=1)
y_train = data.loc[data[‘sample’]==‘train’, ‘isDefault’]