whylogs.mlflow.model_wrapper

Module Contents

Classes

ModelWrapper

Attributes

logger
PyFuncOutput

whylogs.mlflow.model_wrapper.logger

whylogs.mlflow.model_wrapper.PyFuncOutput

class whylogs.mlflow.model_wrapper.ModelWrapper(model)

Bases: object

create_logger(self)

predict(self, data: pandas.DataFrame) → PyFuncOutput

Wrapper around https://www.mlflow.org/docs/latest/_modules/mlflow/pyfunc.html#PyFuncModel.predict This allows us to capture input and predictions into whylogs