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))