1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00

fixed the LSTM middleware initialization

This commit is contained in:
Itai Caspi
2017-12-30 15:18:09 +02:00
committed by Itai Caspi
parent b435c6d2d7
commit eeb3ec5497
4 changed files with 8 additions and 0 deletions

View File

@@ -28,6 +28,8 @@ class ValueOptimizationAgent(Agent):
self.q_values = Signal("Q")
self.signals.append(self.q_values)
self.reset_game(do_not_reset_env=True)
# Algorithms for which q_values are calculated from predictions will override this function
def get_q_values(self, prediction):
return prediction