Jupyter is good for experiments, but create also a python script for end to end training and evaluation