From 408bebce7aac7e492e5975e56c97b2e297806a64 Mon Sep 17 00:00:00 2001
From: PawelDopierala <pawel.dopierala4@gmail.com>
Date: Wed, 12 Jun 2024 15:08:00 +0200
Subject: [PATCH] fix sacred

---
 sacred/create_model.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/sacred/create_model.py b/sacred/create_model.py
index 6d1b781..db590c9 100644
--- a/sacred/create_model.py
+++ b/sacred/create_model.py
@@ -55,11 +55,9 @@ def main(epochs, learning_rate, batch_size, _run):
 
     rmse = np.sqrt(mean_squared_error(Y_test, test_predictions))
     mae = mean_absolute_error(Y_test, test_predictions)
-    r2 = r2_score(Y_test, test_predictions)
 
     _run.log_scalar("rmse", rmse)
     _run.log_scalar("mae", mae)
-    _run.log_scalar("r2", r2)
 
 
 if __name__ == '__main__':