CLVWrapper

class pymc_marketing.mlflow.CLVWrapper(model, method)

Wrapper class for logging with MLflow.

Methods

CLVWrapper.__init__(model, method)
CLVWrapper.load_context(context)	Loads artifacts from the specified PythonModelContext that can be used by predict() when evaluating inputs.
CLVWrapper.predict(context, model_input[, ...])	Perform predictions or sampling using the specified prediction method.
CLVWrapper.predict_stream(context, model_input)	Evaluates a pyfunc-compatible input and produces an iterator of output.

Attributes

predict_type_hints

Internal method to get type hints from the predict function signature.