سلام

الگوریتم شما در لایه‌ی آخر ۱۰ بعدی است در حالی که تارگت بعد ندارد (در واقع صفر بعدی است).

تارگت (y_test, y_train) را قبل از ورود به الگوریتم باید one hot encode کنید. برای مثال 

y_test = to_categorical(y_test)
y_train = to_categorical(y_train)

اطلاعات بیشتر:

https://www.tensorflow.org/api_docs/python/tf/keras/utils/to_categorical