machine-learning - 如何使用腌制文件作为 keras 的数据集

Question

我已经为数字分类构建了自己的数据集，它与 lisa lab 开发的卷积网络模型（这里）配合得很好。我想可视化权重，我想通过 keras 来实现。

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

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

但我希望我的腌制数据集加载而不是 mnist 默认数据。用于 keras 的 mnist 模块在哪里加载它的数据集？而且，我怎样才能传递我自己的数据集而不是使用来自 keras 的 mnist 模块？

提前致谢。

Answer 1

您可以通过阅读源获得有关此方法的大量信息：https ://github.com/fchollet/keras/blob/master/keras/datasets/mnist.py

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

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