张量流中的classifier.predict出错(shape_and_slice规范中的形状不匹配)

Question

我是tensorflow的新手,并且了解我遵循教程的基础知识.复制此代码(https://github.com/random-forests/tutorials/blob/master/ep7.ipynb)并尝试执行它时,我收到一个错误.

W tensorflow/core/framework/op_kernel.cc:993]无效参数:shape_and_slice规范中的形状[1,10]与检查点中存储的形状不匹配:[784,10] Traceback(最近一次调用最后一次):...

似乎负责的命令是以下:

 classifier.predict(test_data[0])

你能告诉我这个问题吗？

Answer 1

请试试

classifier.predict(np.array([test_data[0]], dtype=float), as_iterable=False)

这对我有用