ribs.visualize.qdax_repertoire_heatmap

ribs.visualize.qdax_repertoire_heatmap(repertoire: MapElitesRepertoire, ranges: Collection[tuple[float, float]], *args: P.args, **kwargs: P.kwargs) None[source]

Plots a heatmap of a single-objective QDax MapElitesRepertoire.

Internally, this function converts a MapElitesRepertoire into a CVTArchive and plots it with cvt_archive_heatmap().

Parameters:
repertoire: MapElitesRepertoire

A MAP-Elites repertoire output by an algorithm in QDax.

ranges: Collection[tuple[float, float]]

Upper and lower bound of each dimension of the measure space, e.g. [(-1, 1), (-2, 2)] indicates the first dimension should have bounds \([-1,1]\) (inclusive), and the second dimension should have bounds \([-2,2]\) (inclusive). This is needed since the MapElitesRepertoire does not store measure space bounds.

*args: P.args

Positional arguments to pass to cvt_archive_heatmap().

**kwargs: P.kwargs

Keyword arguments to pass to cvt_archive_heatmap().

Raises:

ValueError – The repertoire passed in has more than one fitness.