Extract targets

metatensor.models.utils.extract_targets.get_outputs_dict(datasets: List[Dataset | Subset])

This is a helper function that extracts all the possible outputs and their gradients from a list of datasets.

Parameters:

datasets (List[Dataset | Subset]) – A list of Datasets or Subsets.

Returns:

A dictionary mapping output names to a list of “values” (always) and possible gradients.