当我打开.py文件时,什么也没发生,但它应该打印预测

2024-04-26 02:45:18 发布

您现在位置:Python中文网/ 问答频道 /正文

当我打开.py文件时,什么也没发生,但它应该打印预测。救命啊? 这是密码。对不起,我英语不好

import tensorflow as tf
from tensorflow import keras

fashion_mnist = keras.datasets.fashion_mnist
(train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()

model = keras.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(128, activation=tf.nn.relu),
    keras.layers.Dense(10, activation=tf.nn.softmax)
])

model.compile(optimizer=tf.train.AdamOptimizer(),
              loss='sparse_categorical_crossentropy')

model.fit(train_images, train_labels, epochs=5)

(test_loss, test_acc) = model.evaluate(test_images, test_labels)

predictions = model.predict(test_images)

print(test_labels)
print(predictions)

Tags: testimportlabelsmodellayerstftensorflowtrain