X_train[train]这一句是什么意思?
代码:
for k, (train, test) in enumerate(kfold):
pipe_lr.fit(X_train[train], y_train[train])
score = pipe_lr.score(X_train[test], y_train[test])
scores.append(score)
print('Fold: %2d, Class dist.: %s, Acc: %.3f' % (k+1,
np.bincount(y_train[train]), score))
代码里用train,test这两个变量,实际是索引切片数组,X_train[train]就是按索引切片取训练集的数据。
具体见官方文档示例:
https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.KFold.html
感觉是从X_train中取出切片train对应数据