train_model.py
import pandas as pd

from sklearn.metrics import f1_score, \ 
                            accuracy_score, \
                            precision_score
from sklearn.model_selection import train_test_split
from sklearn.ensemble import RandomForestClassifier

import mlflow
import mlflow.sklearn

Imports

MLflow imports