Train a machine learning model