Adding missing get_policy and get_value to qlearning

kcelebi · Sep 8, 2024 · 1a622f4 · 1a622f4
1 parent c74c1d4
commit 1a622f4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/machine_learning/q_learning_agents.py b/src/machine_learning/q_learning_agents.py
@@ -80,8 +80,10 @@ def update(self, state, action, next_state, reward):
 		self.q_values[(state, action)] = self.get_Q_value(state, action) + self.alpha * (reward + self.discount*NSQ - self.get_Q_value(state, action))
 
 
+	def get_policy(self, state):
+		return self.compute_action_from_Q_values(state)
 
-
-
+	def get_value(self, state):
+		return self.compute_value_from_Q_values(state)