如何使用pickled文件作为keras的数据集

Question

我已经构建了自己的数字分类数据集，它与lisa lab（Here）开发的卷积网络模型配合得很好。我想要想象权重，我想通过keras来做。

Keras文档试图像这样加载mnist数据：

(X_train, y_train), (X_test, y_test) = mnist.load_data()

但我希望加载pickle数据集而不是mnist默认数据。 kera的mnist模块在哪里加载它的数据集？而且，我如何传递自己的数据集而不是使用keras中的mnist模块？

提前致谢。

Answer 1

通过阅读来源：https://github.com/fchollet/keras/blob/master/keras/datasets/mnist.py

，您可以获得有关此方法的大量信息

在这种情况下，数据集是从Amazon S3存储桶加载的pickle文件。

您可以编写此函数的副本并自行使用它来加载不同的pickle数据集。