Automated model training | Luigi | Python