Machine Learning utilities

MLflow

任意MLflow模型都可以与pandas-on-Spark DataFrame一起使用,前提是它们实现了‘pyfunc’风格。大多数由MLflow支持的框架(如scikit-learn、pytorch、tensorflow等)都是这种情况。更多信息请参见 load_model() 中的综合示例。

注意

必须安装 MLflow 包才能使用此模块。

PythonModelWrapper (model_uri, return_type_hint)

围绕 MLflow 的 Python 对象模型的包装器。

load_model (model_uri[, predict_type])

加载一个MLflow模型到一个包装器中,该包装器可以用于pandas和pandas-on-Spark DataFrame。