Spark mlとPMMLエクスポート

Question

PMMLとSpark-MLlibのモデルをエクスポートすることは可能ですが、Spark-MLについてはどうですか？

それはtoPMML()メソッドを呼び出すことができるようにorg.apache.spark.mllib.regressionからLinearRegressionModelにorg.apache.spark.ml.regressionからLinearRegressionModelを変換することは可能ですか？

Answer 1

あなたはJPMML-SparkMLライブラリを使用してPMMLにスパークMLパイプラインを変換することができます：

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