From 6cb83f36cefdacaf2eb5dc47dc064713fffd4024 Mon Sep 17 00:00:00 2001 From: ferielamira Date: Mon, 20 May 2024 19:32:41 +0100 Subject: [PATCH] "removing enforce exploitatio" --- abm/projects/madrl_foraging/madrl_agent/brain.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/abm/projects/madrl_foraging/madrl_agent/brain.py b/abm/projects/madrl_foraging/madrl_agent/brain.py index 291f897..2679863 100644 --- a/abm/projects/madrl_foraging/madrl_agent/brain.py +++ b/abm/projects/madrl_foraging/madrl_agent/brain.py @@ -130,8 +130,7 @@ def select_action(self, state): if len(self.legal_actions)==1: self.action_tensor = torch.LongTensor([[0]]).to(device) - if 1 in self.legal_actions: - self.action_tensor = torch.LongTensor([[1]]).to(device) + else: # Epsilon-greedy exploration