diff --git a/lab8/trainScript.py b/lab8/trainScript.py
index 5941ccf..10916c9 100644
--- a/lab8/trainScript.py
+++ b/lab8/trainScript.py
@@ -100,13 +100,13 @@ def train():
     # pred = np.array(linear_model.predict(feature_test_sample))
     # flatten_pred = flatten(pred)
     #
-    ## with open(cwd + "/../result.txt", "w+") as resultFile:
+    # with open(cwd + "/../result.txt", "w+") as resultFile:
     #   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_metric('final_loss', min(hist["val_loss"]))
+    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))