2016-01-21 33 views
0

我已經構建了自己的數字分類數據集,並且它與由lisa lab(Here)開發的卷積網絡模型很好地協作。我想要顯示權重,我想通過keras來完成。如何將醃製文件用作keras數據集

Keras文件試圖加載這樣的數據MNIST:

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

但我想我醃的數據集加載,而不是默認MNIST數據。用於keras的mnist模塊在哪裏加載數據集?而且,如何通過我自己的數據集而不是使用keras中的mnist模塊?

在此先感謝。

回答

1

你可以通過閱讀源得到了很多有關此方法的信息的:https://github.com/fchollet/keras/blob/master/keras/datasets/mnist.py

在這種情況下,該數據集是從亞馬遜S3鬥裝泡菜文件。

你可以寫一個這個函數的副本並自己使用它來加載一個不同的pickled數據集。

+0

我已經找到本地文件(/ usr/local/lib/python 2.7/dist-packages/keras/datasets/mnist.py),我可以使用我的數據集位置對其進行修改。 但是,結果是一個問題是「__init__.py」。每個目錄都有__init__.py,但加載mnist數據給出: 'from keras.datasets import mnist ImportError:沒有名爲datasets的模塊 keras目錄已經有一個名爲'datasets'的目錄並且存在__init__.py。我不知道爲什麼這不起作用。 – Anirban