[
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-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
