Defining a machine learning model