機器學習工具

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。