您要让神经网络评估每个输入只有一个输入的784个案例,而不是每个输入有784个输入的单个案例。我有同样的问题,我解决了它有一个包含单个元素的数组,该元素是输入数组。参见下面的示例,第一个起作用,而第二个给出您遇到的错误。
model.predict(np.array([[0.5, 0.0, 0.1, 0.0, 0.0, 0.4, 0.0, 0.0, 0.1, 0.0, 0.0]]))model.predict(np.array([0.5, 0.0, 0.1, 0.0, 0.0, 0.4, 0.0, 0.0, 0.1, 0.0, 0.0]))
希望这也能为您解决:)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)