sklearn 划分测试集和训练集函数区别
1、KFold
from sklearn.model_selection import KFold
kf = KFold(n_splits=3)
X = np.ones(10)
y = [0, 0, 1, 1, 0, 0, 1, 1, 1, 1]
for train, test in kf.split(X):
print("%s %s" % (train, test))
# [4 ...
原创
2019-10-12 14:48:18 ·
743 阅读 ·
1 评论