run_ml function

Run the machine learning pipeline