如何在Keras中将'mnist.load_data()'的返回值格式更改为'mnist_train.csv'?

时间:2017-10-20 09:48:23

标签: numpy keras mnist

我使用Keras。

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

X_train'形状为(number_of_training_sample,224,224,3)

Y_train的形状为(number_of_training_sample, 10)

功能和标签在不同的ndarray中分开。 但我想将这些ndarray更改为'mnist_train.csv'格式。 (https://raw.githubusercontent.com/sjwhitworth/golearn/master/examples/datasets/mnist_train.csv

我自己的数据集与'mnist.load_data()'的返回值完全相同。但只有'mnist_train.csv'格式在我想要使用的代码中工作。

你可以告诉我如何改变格式???? (对不起,对于非常简单的问题..我是numpy和python的新手。)

1 个答案:

答案 0 :(得分:0)

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

将数据加载为numpy数组

'mnist_train.csv'是硬盘上的.csv文件存储,我们通常用pandas库读取

import pandas as pd
X_train = pd.read_csv('filename.csv')

Pandas将.csv作为数据框读取(如果存在,则使用列名称) 使用keras你需要numpy数组,所以

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

应该可以正常工作

发布您正在使用的代码以及为进一步解释而产生的错误