Machine learning model custom train