資料探勘實踐(金融風控)-第四次任務

bigdataworker發表於2020-09-24

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’]

相關文章