nip.utils.malt_forest

nip.utils.malt_forest#

Utilities for reconstructing and visualising a forest of trees from MALT rollouts.

The MALT [MSD+24] trainer samples a set of trees of responses. These are stored in a flat array. This module provides functions to reconstruct the trees from this flat array.

Modules

forest

The data structure and utility for reconstructing a MALT forest.

visualise

Utilities for visualising the MALT forest of trees.