pfnet / pfrl

PFRL: a PyTorch-based deep reinforcement learning library
MIT License
1.18k stars 158 forks source link

Eval stats collector #80

Closed prabhatnagarajan closed 3 years ago

prabhatnagarajan commented 3 years ago

We currently collect basic performance statistics as well as agent statistics. This PR allows environment statistics to be collected. This gives a lot of flexibility. For example, if you have a binary success task such as grasping, your environment could give a "success" statistic, in addition to the reward. If you have a finance domain, your environment could track "profit" separate from the reward function.

The environment optionally implements two functions: get_statistics and clear_statistics. If these functions are not implemented, then the evaluator works as it did before (i.e. this PR is backwards compatible).

prabhatnagarajan commented 3 years ago

/test

pfn-ci-bot commented 3 years ago

Successfully created a job for commit 3326b59:

prabhatnagarajan commented 3 years ago

/test

pfn-ci-bot commented 3 years ago

Successfully created a job for commit 53eaa8a:

prabhatnagarajan commented 3 years ago

/test

pfn-ci-bot commented 3 years ago

Successfully created a job for commit 88ce4cb:

prabhatnagarajan commented 3 years ago

/test

pfn-ci-bot commented 3 years ago

Successfully created a job for commit fe82aba:

prabhatnagarajan commented 3 years ago

/test

pfn-ci-bot commented 3 years ago

Successfully created a job for commit 2b7b0f7: