1个回答
X = np.array([[1, 2], [3, 4], [1, 2], [3, 4]])
test_fold = [0, 1, -1, 1]
test_fold 长度和训练集相同。
-1所在的index只放在train_set里。
其他的数字,每一轮选一个数字,该数字所在的所有index作为test_set。
比如说第一轮,数字0所在的位置作为test_set,即X[0];
第二轮,数字1所在的位置作为test_set,即X[1],X[3]。
SofaSofa数据科学社区DS面试题库 DS面经