如何在本地计算机上训练和测试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)

你能告诉我如何训练和测试CNN的数据文件mnist吗_测试.csv和mnist_火车.csv驻留在我电脑的D:/datafile。在

Answer 1

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

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