Repository: systemml Updated Branches: refs/heads/master db378759e -> 3fa8d3793
[MINOR] Bugfix in mllearn's SVM class - The PR #687 introduced a bug in mllearn SVM class as the scala wrappers were not updated after adding the command-line parameter scoring_only. Project: http://git-wip-us.apache.org/repos/asf/systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/3fa8d379 Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/3fa8d379 Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/3fa8d379 Branch: refs/heads/master Commit: 3fa8d379308ec02971e63f686089d132ad8113d0 Parents: db37875 Author: Niketan Pansare <[email protected]> Authored: Mon Dec 4 19:08:37 2017 -0800 Committer: Niketan Pansare <[email protected]> Committed: Mon Dec 4 19:14:38 2017 -0800 ---------------------------------------------------------------------- src/main/scala/org/apache/sysml/api/ml/SVM.scala | 1 + 1 file changed, 1 insertion(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/systemml/blob/3fa8d379/src/main/scala/org/apache/sysml/api/ml/SVM.scala ---------------------------------------------------------------------- diff --git a/src/main/scala/org/apache/sysml/api/ml/SVM.scala b/src/main/scala/org/apache/sysml/api/ml/SVM.scala index 2013385..c4b7ae4 100644 --- a/src/main/scala/org/apache/sysml/api/ml/SVM.scala +++ b/src/main/scala/org/apache/sysml/api/ml/SVM.scala @@ -105,6 +105,7 @@ class SVMModel(override val uid: String)(estimator: SVM, val sc: SparkContext, v val script = dml(ScriptsUtils.getDMLScript(if (isMultiClass) SVMModel.predictionScriptPathMulticlass else SVMModel.predictionScriptPathBinary)) .in("$X", " ") .in("$model", " ") + .in("$scoring_only", "TRUE") .out("scores") val w = estimator.mloutput.getMatrix("w")
