Skip to content

Commit

Permalink
code clean
Browse files Browse the repository at this point in the history
  • Loading branch information
Felixvillas committed Sep 18, 2022
1 parent 6a40389 commit 2c09cae
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions model.py
Original file line number Diff line number Diff line change
Expand Up @@ -307,8 +307,4 @@ def evaluate(self, env, episode_num=32):
[sum(eval_ep_rewards), len(eval_ep_rewards), 1. if 'battle_won' in info and info['battle_won'] else 0.]
)

# start = episode_num // 4
# end = episode_num * 3 // 4
# sort_eval_data = sorted(eval_data, key=lambda x: x[-1])[start: end] # sorted by win or nor
# return np.mean(sort_eval_data, axis=0)
return np.mean(eval_data, axis=0)

0 comments on commit 2c09cae

Please sign in to comment.