Skip to content

Commit

Permalink
print more digits
Browse files Browse the repository at this point in the history
  • Loading branch information
rizar committed Feb 19, 2019
1 parent 947b01e commit 23f9a38
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion scripts/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@ def main(args, seed, episodes):
num_frames_per_episode = utils.synthesize(logs["num_frames_per_episode"])

if args.model is not None:
print("F {} | FPS {:.0f} | D {} | R:xsmM {:.2f} {:.2f} {:.2f} {:.2f} | S {:.2f} | F:xsmM {:.1f} {:.1f} {} {}"
print("F {} | FPS {:.0f} | D {} | R:xsmM {:.3f} {:.3f} {:.3f} {:.3f} | S {:.3f} | F:xsmM {:.1f} {:.1f} {} {}"
.format(num_frames, fps, duration,
*return_per_episode.values(),
success_per_episode['mean'],
Expand All @@ -97,6 +97,7 @@ def main(args, seed, episodes):
if n > 0:
print("{} worst episodes:".format(n))
for i in indexes[:n]:
print(logs['seed_per_episode'][i])
if args.model is not None:
print("- episode {}: R={}, F={}".format(i, logs["return_per_episode"][i], logs["num_frames_per_episode"][i]))
else:
Expand Down

0 comments on commit 23f9a38

Please sign in to comment.