-
- Downloads
[SPARK-14709][ML] spark.ml API for linear SVM
## What changes were proposed in this pull request? jira: https://issues.apache.org/jira/browse/SPARK-14709 Provide API for SVM algorithm for DataFrames. As discussed in jira, the initial implementation uses OWL-QN with Hinge loss function. The API should mimic existing spark.ml.classification APIs. Currently only Binary Classification is supported. Multinomial support can be added in this or following release. ## How was this patch tested? new unit tests and simple manual test Author: Yuhao <yuhao.yang@intel.com> Author: Yuhao Yang <hhbyyh@gmail.com> Closes #15211 from hhbyyh/mlsvm.
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/LinearSVC.scala 546 additions, 0 deletions.../scala/org/apache/spark/ml/classification/LinearSVC.scala
- mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala 2 additions, 2 deletions...g/apache/spark/ml/classification/LogisticRegression.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/LinearSVCSuite.scala 241 additions, 0 deletions...a/org/apache/spark/ml/classification/LinearSVCSuite.scala
Loading
Please register or sign in to comment