I have a simple model, and I\'m trying to run a classification task.
The following is my model
num_epochs = 10 batch_size = 64 learning_rate = 0.001 input