I am not trying RL yet and still shooting for the low hanging fruits. So basically, trying to train a NN and predict what runs is going to be like