scala - 如何加载 PMML 模型？

Question

我正在按照 PMML model export - spark.mllib 的说明进行操作创建 K-means 模型。

val numClusters = 10
val numIterations = 10
val clusters = KMeans.train(data, numClusters, numIterations)
// Save and load model: export to PMML
println("PMML Model:\n" + clusters.toPMML("/kmeans.xml"))

但我不知道在那之后如何加载 PMML。

我想

val sameModel = KMeansModel.load(sc, "/kmeans.xml")

并出现:

org.apache.hadoop.mapred.InvalidInputException: Input path does not exist: file:/kmeans.xml/metadata

任何的想法？

此致

Answer 1

如文档中所述(对于您似乎感兴趣的版本 - 1.6.1 以及最新可用的 - 2.1.0)Spark 仅支持导出到 PMML。 load方法实际上期望检索以 Spark 自己的格式保存的模型，这就是 load 的原因。方法期望存在某个路径以及抛出异常的原因。

如果您使用 Spark 训练模型，则可以 save它和 load以后吧。

如果您需要加载尚未在 Spark 中训练并已保存为 PMML 的模型，您可以使用 jpmml-spark加载和评估它。