Question

我正在尝试通过训练和测试集将数据拆分为K折。我被困在最后：

我有一个数据集示例：

       [1,2,3,4,5,6,7,8,9,10]

我已经成功创建了用于5倍交叉验证的分区，输出为

     fold=[[2, 1], [6, 0], [7, 8], [9, 5], [4, 3]]

现在，我想创建K个具有K-1训练数据和1个验证集的实例。

我正在使用以下代码：

    ```
      cross_val={"train":[],"test":[]}
       new_fold=folds.copy()
       for i in range(4):
           val=folds.pop(i)
           cross_val["train"].append(folds)
           cross_val["test"].append(val)

           folds[i:i]=[val]```

我得到的输出是：

  {'train': [[[6, 0], [7, 8], [9, 5], [4, 3]],
           [[6, 0], [7, 8], [9, 5], [4, 3]],
          [[6, 0], [7, 8], [9, 5], [4, 3]],
        [[6, 0], [7, 8], [9, 5], [4, 3]]],
   'test': [[6, 0], [7, 8], [9, 5], [4, 3]]}

这是我得到的错误输出。

但是我希望输出为

          train                                   test
          [[6, 0], [7, 8], [9, 5], [4, 3]]      [2,1]
          [[2, 1], [7, 8], [9, 5], [4, 3]]      [6,0]
          [[6, 0], [2, 1], [9, 5], [4, 3]]      [7,8]
          [[6, 0], [7, 8], [9, 5], [2, 1]]       [4,3]
          [[6, 0], [7, 8], [2, 1], [4, 3]]       [9,5]

Answer 1

您每次在此处对相同列表进行编辑，然后将该列表附加多次次。结果，如果您编辑列表，则会在列表的所有元素中看到该编辑。

您可以使用以下方法创建交叉验证：

train = []
test = []
cross_val={'train': train, 'test': test}
for i, testi in enumerate(fold):
    train.append(fold[:i] + fold[i+1:])
    test.append(testi)

对于给定的样本数据，这给我们：

>>> pprint(cross_val)
{'test': [[2, 1], [6, 0], [7, 8], [9, 5], [4, 3]],
 'train': [[[6, 0], [7, 8], [9, 5], [4, 3]],
           [[2, 1], [7, 8], [9, 5], [4, 3]],
           [[2, 1], [6, 0], [9, 5], [4, 3]],
           [[2, 1], [6, 0], [7, 8], [4, 3]],
           [[2, 1], [6, 0], [7, 8], [9, 5]]]}

创建不带sklearn的Kfold交叉验证集

1 个答案: