如何将sparkR模型导出为PMML？

Question

我正在尝试将一个sparkR模型导出为PMML。

第一种方法是使用pmml库：

library(pmml)
sparkR.session()
data(iris)
df <- createDataFrame(iris)
model <- spark.kmeans(df, Sepal_Length ~ Sepal_Width, k = 4, initMode = "random")
model_pmml <- pmml(model)

错误：

Error in UseMethod("pmml"): no applicable method for 'pmml' applied to an object of class "KMeansModel"
Traceback:

1. pmml(model)

我还研究了是否可以在SparkR中使用scala模型上可用的toPMML方法。我找到了一个question，它表明使用Sparklyr可能是可行的，但使用SparkR是不行的。

有什么想法吗？

Answer 1

我得出的结论是不支持导出spark R模型。我为此添加了一个特性请求：https://issues.apache.org/jira/browse/SPARK-21430。如果您也在寻找此功能，请在jira票证上投票。