java - Apache Spark MLlib : How to import model from PMML

Question

我有一个 PMML 文件，它编码了一个不是从 MLlib 导出的逻辑回归模型。

如何使用 Java 中的 MLlib 从 PMML 导入模型以进行评估/预测？

(我知道 MLlib can export to PMML ，但我需要从 PMML 导入)

Answer 1

你可以使用 PMML4S-Spark将 PMML 作为 SparkML 转换器导入，然后在 Scala 中进行预测/评估，例如:

import org.pmml4s.spark.ScoreModel

val model = ScoreModel.fromFile("the/pmml/model/path")
val scoreDf = model.transform(df)

如果你使用 PySpark，你可以使用 PyPMML-Spark ，例如:

from pypmml_spark import ScoreModel

model = ScoreModel.fromFile('the/pmml/model/path')
score_df = model.transform(df)