如何在本地计算机上的文件上训练和测试 tensorflow CNN

Question

我是 python 和 tensorflow 的新手。我正在尝试使用在 tensorflow 中实现的卷积神经网络算法。需要做的是训练和测试我自己的数据集而不是 MNIST 数据集。 当前程序正在使用：

def main(unused_argv):
  # Load training and eval data
  mnist = tf.contrib.learn.datasets.load_dataset("mnist")
  train_data = mnist.train.images  # Returns np.array
  train_labels = np.asarray(mnist.train.labels, dtype=np.int32)
  eval_data = mnist.test.images  # Returns np.array
  eval_labels = np.asarray(mnist.test.labels, dtype=np.int32)

您能否指导我如何使用位于我计算机 D:/datafile 的数据文件 mnist_test.csv 和 mnist_train.csv 训练和测试 CNN。

Answer 1

您可以使用下面的命令通过ronnie.ai framework 来训练和评估模型

eResult = CNNModel.trainNeval(trainData, evalData, designer.modelDef, 200, 50, 1, evalSteps=1)