From ef696cf204aac367110811a8e0ecb81bca58581c Mon Sep 17 00:00:00 2001 From: nicedlp Date: Thu, 7 Sep 2023 08:49:39 +0000 Subject: [PATCH] Add mlFlow logs --- formation_indus_ds_avancee/train_and_predict.py | 1 + 1 file changed, 1 insertion(+) diff --git a/formation_indus_ds_avancee/train_and_predict.py b/formation_indus_ds_avancee/train_and_predict.py index 8055aec..b54aa3b 100644 --- a/formation_indus_ds_avancee/train_and_predict.py +++ b/formation_indus_ds_avancee/train_and_predict.py @@ -19,6 +19,7 @@ def train_model(features: pd.DataFrame, model_registry_folder: str) -> None: y = features[target] with mlflow.start_run(): # insert autolog here ... + mlflow.sklearn.autolog() model = RandomForestRegressor(n_estimators=1, max_depth=10, n_jobs=1) model.fit(X, y) time_str = time.strftime('%Y%m%d-%H%M%S')