如何在遍历数据集时将数据集划分为k倍的折叠数并分配给列表?

时间:2020-04-25 06:18:42

标签: python-3.x list cross-validation k-fold

考虑列表:

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

我想将列表分为3个不同的列表:

list_1 = [1, 2, 3]
list_2 = [4, 5, 6]
list_3 = [7, 8, 9]

有没有一种方法可以划分列表并将其元素分配给range(len(lst_main))中的i的list_i。

即我只想在遍历循环到list_i时分配元素,这样当i = 1时,list_1就可以了。当i = 10时,列表的名称应为list_10。

for i in range(len(lst_main)):
    list_i = lst_main[:i]

要解决的问题是实现K倍交叉验证,其中给定了一个数据集,并且必须将数据集除以k倍数。因此,如果数据集是一个长列表,且折叠倍数k = 5,则必须将数据集分为5个相等的部分。

因此,在遍历数据集,将数据集划分为5个相等的部分并将第一部分分配给list1并将第二部分分配给list2的过程中,依此类推。

0 个答案:

没有答案
相关问题