在Python中以任意块分割矩阵(或数组)的优雅方法

2024-04-19 02:10:21 发布

您现在位置:Python中文网/ 问答频道 /正文

我在一个数据集上做k倍交叉验证,为此我需要将我的数据分成两部分,一部分是训练集,另一部分是测试集。你知道吗

到目前为止,我已经成功地将数据集拆分为我的测试集:

testindices = range(idx*testsize, (idx*testsize) + testsize) 
testset = dataset[testindices,0:numfeatures] 

其中idx是第k次折叠,或者换句话说是块起始索引。这很简单。我被困在如何得到另一部分,即车列?它基本上是一个集不同的操作,由此产生的列车集是所有除了测试集?你知道吗


Tags: 数据range交叉dataset集上idx列车testset