change for r2

VIDA-NYU · May 1, 2024 · 5bf1514 · 5bf1514
1 parent 90c71b0
commit 5bf1514
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/alpha_automl/pipeline_search/agent_environment.py b/alpha_automl/pipeline_search/agent_environment.py
@@ -94,7 +94,8 @@ def step(self, action):
         if game_end == 1:  # pipeline score over threshold
             try:
                 if self.game.problem == "REGRESSION":
-                    reward = 10 + (100 / self.game.getEvaluation(self.board))
+                    # reward = 10 + (100 / self.game.getEvaluation(self.board))
+                    reward = 10 + (self.game.getEvaluation(self.board)) ** 2 * 100
                 else:
                     reward = 10 + (self.game.getEvaluation(self.board)) ** 2 * 100
             except Exception as e: