# AI setup
N = 20
batch_size = 5
n_epochs = 4
alpha = 0.0003