Я только начал использовать Tensorflow, и у меня возникла ошибка.
cifar10_mnist = keras.datasets.cifar10
(train_images, train_labels), (test_images, test_labels) = cifar10_mnist.load_data()
train_images = train_images/255
test_images = test_images/255
classes = ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship',
'truck']
model = keras.Sequential([
keras.layers.Flatten(input_shape=(32,32)),
keras.layers.Dense(150, activation="relu"),
keras.layers.Dense(10, activation="softmax")
])
model.compile(optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
model.fit(train_images, train_labels, epochs=5)
test_loss, test_acc = model.evaluate(test_images, test_labels)
prediction = model.predict(test_images)
answer = np.argmax(prediction[0])
print(classes[answer])
Я не совсем уверен, как это сделать, я оглядывался назад, но не вижу ничего, что я мог бы изменить, чтобы это исправить. Спасибо, очень ценю помощь. :)