jpmml / jpmml-sparkml

Java library and command-line application for converting Apache Spark ML pipelines to PMML
GNU Affero General Public License v3.0
267 stars 80 forks source link

Can I convert model to pmml? #85

Closed Mantj closed 4 years ago

Mantj commented 4 years ago

I have a GBTClassificationModel, and I want to save it as pmml, but not put it into pipeline, what can I do? thanks~

vruusmann commented 4 years ago

The pipeline wrapper is required, because it holds all sorts of supporting metadata about the learning process (eg. column names).

It should be possible to bypass this "convenience layer", and work directly with lower JPMML-SparkML library layers and APIs. However, I don't have the time to guide you through there.