java - Spark ml 和 PMML 导出

Question

我知道可以像PMMLwith一样导出模型Spark-MLlib，但是呢Spark-ML？

是否可以将LinearRegressionModelfrom转换org.apache.spark.ml.regression为LinearRegressionModelfromorg.apache.spark.mllib.regression以便能够调用该toPMML()方法？

Answer 1

您可以使用JPMML-SparkML库将 Spark ML 管道转换为 PMML ：

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));