Feat: Add custom legends to plots - Githubissues

instadeepai / marl-eval

A tool for aggregating and plotting MARL experiment data.

https://sites.google.com/view/marl-standard-protocol/home

Apache License 2.0

57 stars 6 forks source link

Feat: Add custom legends to plots #43

Closed RuanJohn closed 7 months ago

RuanJohn commented 8 months ago

What?

Add support for passing in a custom legend mapping dictionary which maps algorithm names in the dataset to names to be displayed on plot legends.

Why?

It makes the plot legends look a lot nicer.

Extra

Added a small epsilon value to the normalisation in the data processing. This is to avoid accidentally dividing by 0.
Updated the tests to account for this.
Updated the example python file to use legend mapping and some of the plots in the quickstart notebook.