Прежде всего, насколько я вижу, ваш код верен. Ваша модель, предсказывающая неправильный элемент, может быть вызвана тем, что модель недостаточно долго обучалась. Я настоятельно рекомендую вам установить epochs=100
, и вы сможете увидеть повышение точности модели. Как правило, вы должны всегда стараться дать вашей модели как можно больше epochs
для обучения. Это просто займет некоторое время. Попробуйте несколько разных чисел epochs
, чтобы найти тот, который не занимает много времени, но все же дает приемлемый результат.