小编Abd*_*chi的帖子

我编写了下面的代码，它假设加载一个模型，然后对 MNIST 数据集中的元素进行预测运行。在执行开始时，代码工作正常，我得到了我想要的预测，但突然我确实得到了以下错误，我不确定这是否与.predict arguments.

我的代码：

# importing libraries
import tensorflow as tf  # deep learning library. Tensors are just multi-dimensional arrays
import gzip,sys,pickle # dataset manipulation library

# importing MNIST dataset
f = gzip.open('mnist.pkl.gz', 'rb')
if sys.version_info < (3,):
    data = pickle.load(f)
else:
    data = pickle.load(f, encoding='bytes')
f.close()
(x_train, _), (x_test, _) = data

print("-----------------------dataset ready-----------------------")
# using an expample from x_test / to remove later
# preprocessing
x_test = tf.keras.utils.normalize(x_test, axis=1)  # scales data between 0 and …