# Hyper-parameters training_epochs = 400 # Total number of training epochs learning_rate = 0.03 # The learning rate