ML problem formulation