假设n_repeats=5
且折叠数为3 (n_splits=3
)。
这是否意味着验证器为我们的估计器/模型创建了3个折叠,每个折叠都供模型使用(就像KFold的作用),然后重复这个过程5次?
这意味着我们的模型总共将使用5 x 3 = 15个折叠?
回答:
是的,你可以通过在循环中调用KFolds.split()
n_repeats
次来基本达到同样的效果。
示例设置:
X = np.array([[1, 2], [3, 4], [1, 2], [3, 4]])y = np.array([0, 0, 1, 1])
然后运行:
rkf = RepeatedKFold(n_splits=2, n_repeats=1, random_state=2652124)for train_index, test_index in rkf.split(X): print("TRAIN:", train_index, "TEST:", test_index)
… 产生的结果是:
TRAIN: [0 1] TEST: [2 3]TRAIN: [2 3] TEST: [0 1]
… 就像KFold(n_splits=2, random_state=2652124)
会产生的效果一样。将n_repeats
改为2时,产生的结果是:
TRAIN: [0 1] TEST: [2 3]TRAIN: [2 3] TEST: [0 1]TRAIN: [1 2] TEST: [0 3]TRAIN: [0 3] TEST: [1 2]
依此类推。