I\'m using Keras\' functional API to build a simple sequential neural net. Here\'s the shape of X_train and y_train_encoded (one-hot-encoded y_train with 10 classes).