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)

您能否指导我如何在数据文件mnist_test.csv和mnist_train.csv上训练和测试CNN，该数据文件位于我的计算机的D：/ datafile中。

Answer 1

您可以使用以下命令使用ronnie.ai framework

训练和评估模型

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

如何在驻留在本地计算机上的文件上训练和测试张量流CNN

1 个答案: