Github user manishamde commented on a diff in the pull request: https://github.com/apache/spark/pull/1290#discussion_r19717451 --- Diff: docs/mllib-ann.md --- @@ -0,0 +1,223 @@ +--- +layout: global +title: Artificial Neural Networks - MLlib +displayTitle: <a href="mllib-guide.html">MLlib</a> - Artificial Neural Networks +--- + +# Introduction + +This document describes the MLlib's Artificial Neural Network (ANN) implementation. + +The implementation currently consist of the following files: + +* 'ArtificialNeuralNetwork.scala': implements the ANN +* 'ANNSuite': implements automated tests for the ANN and its gradient +* 'ANNDemo': a demo that approximates three functions and shows a graphical representation of +the result + +# Summary of usage + +The "ArtificialNeuralNetwork" object is used as an interface to the neural network. It is +called as follows: + +``` +val annModel = ArtificialNeuralNetwork.train(rdd, hiddenLayersTopology, maxNumIterations) --- End diff -- How common is it to have different number of nodes in the hidden layer? I am wondering whether there should be support for a simpler method ```def train(rdd, numNodesHiddenLayers, maxNumIterations)``` and perhaps even a simpler ```def train(rdd)``` with good default settings to help users get started. @mengxr @jkbradley Would the upcoming MLlib API feature make this suggestion moot with support for default parameters?
--- If your project is set up for it, you can reply to this email and have your reply appear on GitHub as well. If your project does not have this feature enabled and wishes so, or if the feature is enabled but not working, please contact infrastructure at infrastruct...@apache.org or file a JIRA ticket with INFRA. --- --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org