机器学习工具

MLflow

任意MLflow模型可以用于pandas-on-Spark Dataframes,只要他们实现“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-on-Spark DataFrame。