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

Create a dataset using an agent (#306)

Browse Source 
Generate a dataset using an agent (allowing to select between this and a random dataset)
This commit is contained in:
Gal Leibovich
2019-05-28 09:34:49 +03:00
committed by GitHub
parent 342b7184bc
commit 9e9c4fd332
26 changed files with 351 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/agents/policy_optimization_agent.py
View File

@@ -147,7 +147,7 @@ class PolicyOptimizationAgent(Agent):
if isinstance(self.spaces.action, DiscreteActionSpace):
# DISCRETE
action_probabilities = np.array(action_values).squeeze()
action = self.exploration_policy.get_action(action_probabilities)
action, _ = self.exploration_policy.get_action(action_probabilities)
action_info = ActionInfo(action=action,
all_action_probabilities=action_probabilities)