首页 > 解决方案 > 使用苏打水训练和部署模型

问题描述

如图所示,我想使用 mlfow spark 训练和部署 h2o 模型:https ://res.infoq.com/presentations/mlflow-databricks/en/slides/sl21-1566324281761.jpg 我发现了两种训练方法型号:1. https://docs.databricks.com/_static/notebooks/h2o-sparkling-water-python.html 2. https://www.h2o.ai/blog/h2os-automl-in-spark

使用第一种方法,我能够训练 H2o 模型,但无法使用 mlflow spark 进行部署,因为它需要一个 spark 模型。当我尝试使用第二种方法时,在运行链接中给出的示例示例时,它会在训练期间引发以下错误。“pyspark.sql.utils.AnalysisException:“无法解析' label'给定的输入列:”

任何人都可以帮助并让我知道我应该使用哪种方法在苏打水中训练模型并在火花中部署相同的方法。

标签: apache-sparkpysparkh2o

解决方案


推荐阅读