Using MLFlow to Track and Version Machine Learning Models

A step-by-step coding practice

When we tune parameters of a machine learning model, we might need to train it for multiple times in order to pick the best model. If the number of trainings goes to too many, we might have two problems.

• How do we track the parameters and metrics of each model? Writing them manually to an excel file? That would be tedious and error-prone.
• How do we version each model? Save them to the disk with a different name for each? It would…