From bf499f6f0967ea5b684e44778bc868bf5ad88152 Mon Sep 17 00:00:00 2001 From: s444417 Date: Sat, 14 May 2022 14:55:29 +0200 Subject: [PATCH] comment loss --- lab8/trainScript.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lab8/trainScript.py b/lab8/trainScript.py index 48cfbc9..632cf90 100644 --- a/lab8/trainScript.py +++ b/lab8/trainScript.py @@ -106,7 +106,7 @@ def train(): mlflow.log_param('epochs', numberOfEpochParam) mlflow.log_param('learning_rate', learning_rate) - mlflow.log_metric('final_loss', min(hist["val_loss"])) + # mlflow.log_metric('final_loss', min(hist["val_loss"])) signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))