diff --git a/docs/mllib-guide.md b/docs/mllib-guide.md index f8e879496c..de7d66fb2d 100644 --- a/docs/mllib-guide.md +++ b/docs/mllib-guide.md @@ -39,6 +39,7 @@ filtering, dimensionality reduction, as well as underlying optimization primitiv * [Optimization (developer)](mllib-optimization.html) * stochastic gradient descent * limited-memory BFGS (L-BFGS) +* [PMML model export](mllib-pmml-model-export.html) MLlib is under active development. The APIs marked `Experimental`/`DeveloperApi` may change in future releases, diff --git a/docs/mllib-pmml-model-export.md b/docs/mllib-pmml-model-export.md new file mode 100644 index 0000000000..42ea2ca81f --- /dev/null +++ b/docs/mllib-pmml-model-export.md @@ -0,0 +1,86 @@ +--- +layout: global +title: PMML model export - MLlib +displayTitle: MLlib - PMML model export +--- + +* Table of contents +{:toc} + +## MLlib supported models + +MLlib supports model export to Predictive Model Markup Language ([PMML](http://en.wikipedia.org/wiki/Predictive_Model_Markup_Language)). + +The table below outlines the MLlib models that can be exported to PMML and their equivalent PMML model. + +
MLlib model | PMML model |
---|---|
KMeansModel | ClusteringModel | +
LinearRegressionModel | RegressionModel (functionName="regression") | +
RidgeRegressionModel | RegressionModel (functionName="regression") | +
LassoModel | RegressionModel (functionName="regression") | +
SVMModel | RegressionModel (functionName="classification" normalizationMethod="none") | +
Binary LogisticRegressionModel | RegressionModel (functionName="classification" normalizationMethod="logit") | +