comment all params

2022-05-14 14:58:46 +02:00 · 2022-05-14 14:58:46 +02:00 · bc8d6c5b70
commit bc8d6c5b70
parent bf499f6f09
1 changed files with 2 additions and 2 deletions
--- a/lab8/trainScript.py
+++ b/lab8/trainScript.py
@ -104,8 +104,8 @@ def train():
    #   resultFile.write("predictions: " + str(flatten_pred) + '\n')
    #   resultFile.write("expected: " + str(labels_test_sample.to_numpy()))

-    mlflow.log_param('epochs', numberOfEpochParam)
-    mlflow.log_param('learning_rate', learning_rate)
+    # mlflow.log_param('epochs', numberOfEpochParam)
+    # mlflow.log_param('learning_rate', learning_rate)
    # mlflow.log_metric('final_loss', min(hist["val_loss"]))

  signature = mlflow.models.signature.infer_signature(house_price_features, linear_model.predict(house_price_features))