[ https://issues.apache.org/jira/browse/SPARK-4752?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14234855#comment-14234855
]
Alexander Ulanov commented on SPARK-4752:
-----------------------------------------
The initial implementation can be found here: https://github.com/avulanov/spark/tree/annclassifier.
It codes the class label as a binary vector in the ANN output and selects the class based
on biggest output value. The implementation contains unit tests as well.
The mentioned code uses the following PR: https://github.com/apache/spark/pull/1290. It is
not yet merged into the main branch. I think that I should not make a pull request until then.
> Classifier based on artificial neural network
> ---------------------------------------------
>
> Key: SPARK-4752
> URL: https://issues.apache.org/jira/browse/SPARK-4752
> Project: Spark
> Issue Type: New Feature
> Components: MLlib
> Affects Versions: 1.1.0
> Reporter: Alexander Ulanov
> Fix For: 1.3.0
>
> Original Estimate: 168h
> Remaining Estimate: 168h
>
> Implement classifier based on artificial neural network (ANN). Requirements:
> 1) Use the existing artificial neural network implementation https://issues.apache.org/jira/browse/SPARK-2352,
https://github.com/apache/spark/pull/1290
> 2) Extend MLlib ClassificationModel trait,
> 3) Like other classifiers in MLlib, accept RDD[LabeledPoint] for training,
> 4) Be able to return the ANN model
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org
|