MMMWrapper.predict#

MMMWrapper.predict(context, model_input, params=None)[source]#

Perform predictions or sampling using the specified prediction method.

Parameters:

contextAny: The context in which the model is running. Isn’t specified by users but is passed by MLflow.
model_inputarray, shape (n_pred, n_features): The input data used for prediction.
paramsdict, optional: A dictionary of parameters to specify the prediction method.

Returns:

ndarray or InferenceData: The predictions or samples generated by the model.

Raises:

ValueError: If an unsupported prediction method is specified.