Remove blackout reward

thatguy11325 · Feb 26, 2024 · 6ec6506 · 6ec6506
1 parent 432e8ac
commit 6ec6506
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/pokemonred_puffer/environment.py b/pokemonred_puffer/environment.py
@@ -949,7 +949,6 @@ def get_game_state_reward(self, print_stats=False):
             "stats_menu": self.seen_stats_menu * 0.1,
             "bag_menu": self.seen_bag_menu * 0.1,
             "cancel_bag_menu": self.seen_cancel_bag_menu * 0.1,
-            "blackout_check": self.blackout_check * 0.0001,
         }
 
         return state_scores

diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py
@@ -227,11 +227,11 @@ def train(args, env_module, make_env):
             "coords": 0.9995,
             "map_ids": 0.995,
             "explore": 0.9995,
-            "start_menu": 0.95,
-            "pokemon_menu": 0.95,
-            "stats_menu": 0.99,
-            "bag_menu": 0.95,
-            "cancel_bag_menu": 0.99,
+            "start_menu": 0.98,
+            "pokemon_menu": 0.98,
+            "stats_menu": 0.98,
+            "bag_menu": 0.98,
+            "cancel_bag_menu": 0.98,
         },
         "forgetting_frequency": 10,
         "perfect_ivs": parsed_args.perfect_ivs,