gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

Add is_on_policy property to agents (#480)

Browse Source
This commit is contained in:
Guy Jacob
2021-05-06 18:02:02 +03:00
committed by GitHub
parent 06bacd9de0
commit 9106b69227
21 changed files with 86 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/agents/qr_dqn_agent.py
View File

@@ -67,6 +67,10 @@ class QuantileRegressionDQNAgent(ValueOptimizationAgent):
super().__init__(agent_parameters, parent)
self.quantile_probabilities = np.ones(self.ap.algorithm.atoms) / float(self.ap.algorithm.atoms)
@property
def is_on_policy(self) -> bool:
return False
def get_q_values(self, quantile_values):
return np.dot(quantile_values, self.quantile_probabilities)