create ml model by using dag - View it on GitHub