Python:如何保存训练数据集

Question

我有训练数据集，分别是 xtrain、ytrain、xtest 和 ytest。它们都是 numpy 数组。我想将它们一起保存到一个文件中，以便我可以将它们加载到工作区中，就像在 keras 中为 mnist.load_data 所做的那样:

(xtrain, ytrain), (xtest, ytest) = mnist.load_data(filepath)

在 python 中，有什么方法可以将我的训练数据集保存到这样一个文件中吗？或者有什么其他的欣赏方法可以挽救吗？

Answer 1

您有多种选择:

• npz
• hdf5
• pickle

Keras 提供了将模型保存到 hdf5 的选项。另请注意，在这三种格式中，它是唯一可互操作的格式。