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