Repository: spark
Updated Branches:
  refs/heads/branch-1.6 a85a9122f -> a91d21314


[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 <fazl...@wso2.com>

Closes #9558 from fazlan-nazeem/master.

(cherry picked from commit 9b88e1dcad6b5b14a22cf64a1055ad9870507b5a)
Signed-off-by: Xiangrui Meng <m...@databricks.com>


Project: http://git-wip-us.apache.org/repos/asf/spark/repo
Commit: http://git-wip-us.apache.org/repos/asf/spark/commit/a91d2131
Tree: http://git-wip-us.apache.org/repos/asf/spark/tree/a91d2131
Diff: http://git-wip-us.apache.org/repos/asf/spark/diff/a91d2131

Branch: refs/heads/branch-1.6
Commit: a91d213146d2dde82710c55a630a3c79f6b7af41
Parents: a85a912
Author: fazlan-nazeem <fazl...@wso2.com>
Authored: Mon Nov 9 08:58:55 2015 -0800
Committer: Xiangrui Meng <m...@databricks.com>
Committed: Mon Nov 9 08:59:02 2015 -0800

----------------------------------------------------------------------
 .../scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala  | 1 +
 1 file changed, 1 insertion(+)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/spark/blob/a91d2131/mllib/src/main/scala/org/apache/spark/mllib/pmml/export/PMMLModelExport.scala
----------------------------------------------------------------------
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 c5fdecd..9267e6d 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 = {


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@spark.apache.org
For additional commands, e-mail: commits-h...@spark.apache.org

Reply via email to