Привет, у меня возникла проблема с этим кодом для предсказания меток изображений из набора данных fashion-mnist. Все прошло хорошо, ошибок нет, но результат оказался не таким, как я ожидал.
Вот важная часть кода, который я написал
(X_train,y_train),(X_test,y_test)=ks.datasets.fashion_mnist.load_data()
X_train=X_train/255
X_test=X_test/255
input_data_shape=(28,)
model=ks.Sequential()
model.add(ks.layers.Flatten(input_shape=input_data_shape,name="input_layer"))
model.add(ks.layers.Dense(32,activation="relu",name="hidden_layer"))
model.add(ks.layers.Flatten())
model.add(ks.layers.Dense(10,activation="softmax",name="output_layer"))
optimizer="adam"
loss_function="sparse_categorical_crossentropy"
metric=["accuracy"]
model.compile(optimizer=optimizer,loss=loss_function,metrics=metric)
model.fit(X_train,y_train,epochs=10)
pred=model.predict(X_test[348])
pred=np.argmax(pred,axis=1)
print(pred)
print(y_test[348])
После запуска кода Я не вижу ошибок, но результат оказался не тем, чего я ожидал. Вот результат:
[0 0 0 0 5 8 1 1 1 1 1 1 1 1 1 1 1 1 1 5 1 1 8 1 1 0 0 0]
9
Почему отображается массив из 28 вместо одной метки? я сделал какую-нибудь ошибку или это должно быть так? пожалуйста, помогите, спасибо.