Я следую учебному пособию по глубокому обучению, и я использовал Jupyter Notebook, чтобы следовать за ним. Программа работала без проблем. Однако, когда я скопировал свой код в PyCharm после установки всех необходимых библиотек, я запустил код и в самом конце, после прохождения обучения, появилась эта ошибка: AttributeError: объект list не имеет атрибута shape. Кто-нибудь может сказать мне, почему эта ошибка возникает и как ее исправить?
Ссылка на учебник: https://www.youtube.com/watch?v=wQ8BIBpya2k&t=1s
Код:
import tensorflow as tf
import numpy as np
fashion_mnist = tf.keras.datasets.fashion_mnist
(x_train, y_train), (x_test, y_test) = fashion_mnist.load_data()
x_train = tf.keras.utils.normalize(x_train, axis=1)
x_test = tf.keras.utils.normalize(x_test, axis=1)
model = tf.keras.models.Sequential()
model.add(tf.keras.layers.Flatten())
model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(10, activation=tf.nn.softmax))
model.compile(optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"])
model.fit(x_train, y_train, epochs=20)
model.save('epic_clothes_identifier')
new_model = tf.keras.models.load_model("epic_clothes_identifier")
predictions = new_model.predict([x_test])
print(np.argmax(predictions[0]))