无法将“SparkXGBRegressorModel”记录到 Databricks 上的功能存储

Question

我正在 Azure Databricks 上开发一个 MLOps 项目，在使用“SparkXGBRegressorModel”库训练模型后，我想将其记录到功能存储中。问题是，使用 fs.log_model() 函数会因模型数据类型为“SparkXGBRegressorModel”而引发错误。

有谁知道如何将此模型登录到Feature Store？

我尝试使用在 Databricks 文档上找到的代码将其注册到功能存储：

fs.log_model(
            model=xgb_model,
            artifact_path=f"{model_name}_artifact_path",
            flavor=mlflow.xgboost,
            training_set = training_set ,
            registered_model_name = model_name
        ) 

但正如我所说，它引发了以下错误：

AttributeError：“SparkXGBRegressorModel”对象没有属性“save_model”

Answer 1

您尝试过：flavor=mlflow.spark吗？