Spark ml和PMML导出 [英] Spark ml and PMML export

问题描述

我知道可以使用 Spark-MLlib 将模型导出为 PMML ，但是<$ c呢$ c> Spark-ML ？

I know that it's possible to export models as PMML with Spark-MLlib, but what about Spark-ML?

是否可以从<转换 LinearRegressionModel code> org.apache.spark.ml.regression 从 org.apache.spark到 LinearRegressionModel 。 mllib.regression 能够调用 toPMML（）方法吗？

Is it possible to convert LinearRegressionModel from org.apache.spark.ml.regression to a LinearRegressionModel from org.apache.spark.mllib.regression to be able to invoke the toPMML() method?

推荐答案

您可以使用 JPMML-SparkML library：

You can convert Spark ML pipelines to PMML using the JPMML-SparkML library:

StructType schema = dataFrame.schema()
PipelineModel pipelineModel = pipeline.fit(dataFrame);
org.dmg.pmml.PMML pmml = org.jpmml.sparkml.ConverterUtil.toPMML(schema, pipelineModel);
JAXBUtil.marshalPMML(pmml, new StreamResult(System.out));

这篇关于Spark ml和PMML导出的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！