MMMWrapper.predict

MMMWrapper.predict(context, model_input, params=None)

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_inputpd.DataFrame, 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.