Multi class

class MultiClassPredictionLabel(
    prediction_scores: Dict[str, Union[float, int]] #required
    threshold_scores: Dict[str, Union[float, int]] = None # optional but required for multi-label use cases
)

class MultiClassActualLabel(
    actual_scores: Dict[str, Union[float, int]] 
)

pred_label = MultiClassPredictionLabel(
	prediction_scores=record["prediction_scores"],
        threshold_scores=record["threshold_scores"],     #additional parameter for multi-label use cases
)
actual_label = MultiClassActualLabel(
	actual_scores={record["actual_class"]: 1},
)
response = arize_client.log(
    model_id="multiclass-classification-multi-label-single-record-ingestion-tutorial",
    model_version="1.0",
    model_type=ModelTypes.MULTI_CLASS,
    environment=Environments.PRODUCTION,
    prediction_id=record["prediction_id"],
    prediction_label=pred_label,
    actual_label=actual_label,
    features=record["features"]
)