From 9180f1e297300325b384dd437c70c0dba8013920 Mon Sep 17 00:00:00 2001 From: YuriCat Date: Sat, 19 Feb 2022 08:42:18 +0900 Subject: [PATCH] feature: output whole result --- handyrl/train.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/handyrl/train.py b/handyrl/train.py index 6f6d56b6..7d2fe4b8 100755 --- a/handyrl/train.py +++ b/handyrl/train.py @@ -467,7 +467,8 @@ def feed_episodes(self, episodes): if episode is None: continue for p in episode['args']['player']: - model_id = episode['args']['model_id'][p] + #model_id = episode['args']['model_id'][p] + model_id = self.model_epoch outcome = episode['outcome'][p] n, r, r2 = self.generation_results.get(model_id, (0, 0, 0)) self.generation_results[model_id] = n + 1, r + outcome, r2 + outcome ** 2 @@ -495,7 +496,8 @@ def feed_results(self, results): if result is None: continue for p in result['args']['player']: - model_id = result['args']['model_id'][p] + #model_id = result['args']['model_id'][p] + model_id = self.model_epoch res = result['result'][p] n, r, r2 = self.results.get(model_id, (0, 0, 0)) self.results[model_id] = n + 1, r + res, r2 + res ** 2