Create higgsml-train.py