Tips for traning an ML model