java - Spark ml と PMML のエクスポート

Question

PMMLと同様にモデルをエクスポートできることは知っていますSpark-MLlibが、ではどうSpark-MLでしょうか。

メソッドを呼び出せるようにfrom からLinearRegressionModelfromorg.apache.spark.ml.regressionに変換することは可能ですか?LinearRegressionModelorg.apache.spark.mllib.regressiontoPMML()

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