如何挑选和挑选多个火车和测试样品?

时间:2014-07-10 15:47:15

标签: numpy pickle python-3.4

我想生成随机序列和测试样本,我已经有样本了。然后,我想挑选每个列车测试集(存储在一个名为Data的类中),这样我就可以单独解开每个测试和训练集来进行实验。

这是我的代码(我正在挑选一个numpy数组):

for i in range(100):
        data = Data()
        data.train_test(100, 11, data_views, class_vals, col_idx)
        data.save('Samples')

然后:

data.read('Samples')

这是我班上的功能:

def save (self, filename):
    with open(filename, 'wb') as f:
        pickle.dump(self, f)

def read (self, filename):
    with open(filename, 'rb') as f:
        rd = pickle.load(f)
    return (rd)

我将如何分别检索已保存的火车和测试集?我查看过pickle文档,但似乎无法找到任何内容。

1 个答案:

答案 0 :(得分:0)

我通过单独保存每个文件而不是一起保存在一个大文件中解决了这个问题,这样我就可以使用for循环分别读取每个文件,对它们进行操作,然后附加所有输出。