autointent.context.optimization_info.DecisionArtifact

class autointent.context.optimization_info.DecisionArtifact(/, **data)

Bases: Artifact

Artifact containing outputs from the predictor node.

Outputs from the best predictor, numpy array of shape (n_samples,) or (n_samples, n_classes) depending on classification mode (multi-class or multi-label)

Parameters:

data (Any)

model_config

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

labels: numpy.typing.NDArray[numpy.float64]