diff options
author | fazlan-nazeem <fazlann@wso2.com> | 2015-11-09 08:58:55 -0800 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-11-09 08:58:55 -0800 |
commit | 9b88e1dcad6b5b14a22cf64a1055ad9870507b5a (patch) | |
tree | 1ffcde006be01ddc39b02d9c92024b1e31b40675 /mllib/src/main | |
parent | d50a66cc04bfa1c483f04daffe465322316c745e (diff) | |
download | spark-9b88e1dcad6b5b14a22cf64a1055ad9870507b5a.tar.gz spark-9b88e1dcad6b5b14a22cf64a1055ad9870507b5a.tar.bz2 spark-9b88e1dcad6b5b14a22cf64a1055ad9870507b5a.zip |
[SPARK-11582][MLLIB] specifying pmml version attribute =4.2 in the root node of pmml model
The current pmml models generated do not specify the pmml version in its root node. This is a problem when using this pmml model in other tools because they expect the version attribute to be set explicitly. This fix adds the pmml version attribute to the generated pmml models and specifies its value as 4.2.
Author: fazlan-nazeem <fazlann@wso2.com>
Closes #9558 from fazlan-nazeem/master.
Diffstat (limited to 'mllib/src/main')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala b/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala index c5fdecd3ca..9267e6dbdb 100644 --- a/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala +++ b/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala @@ -32,6 +32,7 @@ private[mllib] trait PMMLModelExport { @BeanProperty val pmml: PMML = new PMML + pmml.setVersion("4.2") setHeader(pmml) private def setHeader(pmml: PMML): Unit = { |