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)

labels

Predicted labels for the samples.

model_config

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

labels: autointent.custom_types.ListOfLabelsWithOOS