ribs.visualize.qdax_repertoire_heatmap

ribs.visualize.qdax_repertoire_heatmap(repertoire, ranges, *args, **kwargs)[source]

Plots a heatmap of a QDax MapElitesRepertoire.

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

Parameters
  • repertoire (qdax.core.containers.mapelites_repertoire.MapElitesRepertoire) – A MAP-Elites repertoire output by an algorithm in QDax.

  • ranges (array-like of (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 – Positional arguments to pass to cvt_archive_heatmap().

  • **kwargs – Keyword arguments to pass to cvt_archive_heatmap().