pred = model.predict(X_test[0].reshape(1, 28, 28, 1)) # 预测测试集第一个数据 print(pred[0], "转换一下格式得到：", pred.argmax()) # 把one-hot编码转换为数字 import matplotlib.pyplot as plt # 导入绘图工具包 plt.imshow(X_test[0].reshape(28, 28), cmap='Greys') # 输出这个图片
