Skip to content

Commit

Permalink
Update ArtificialNeuralNetwork.scala
Browse files Browse the repository at this point in the history
Fix ScalaStyle
  • Loading branch information
bgreeven committed Nov 3, 2014
1 parent c68c519 commit 73759d5
Showing 1 changed file with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ class ArtificialNeuralNetwork private[mllib](
setConvergenceTol(convergenceTol).
setMaxNumIterations(maxNumIterations)

/**
/**
* Trains the ANN model.
* Uses default convergence tolerance 1e-4 for LBFGS.
*
Expand Down Expand Up @@ -304,10 +304,10 @@ object ArtificialNeuralNetwork {
* @return random weights vector.
*/
def randomWeights(
inputLayerSize: Int,
outputLayerSize: Int,
hiddenLayersTopology: Array[Int],
seed: Int): Vector = {
inputLayerSize: Int,
outputLayerSize: Int,
hiddenLayersTopology: Array[Int],
seed: Int): Vector = {
val topology = inputLayerSize +: hiddenLayersTopology :+ outputLayerSize
return randomWeights(topology, true, seed)
}
Expand Down

0 comments on commit 73759d5

Please sign in to comment.