search

cloudml / zen

Zen aims to provide the largest scale and the most efficient machine learning platform on top of Spark, including but not limited to logistic regression, latent dirichilet allocation, factorization machines and DNN.
Apache License 2.0
170 stars 75 forks source link

Add convergenceTol #53

Open benmccann opened 8 years ago

benmccann commented 8 years ago

Spark's GradientDescent optimizer has a convergenceTol which is very helpful. It would be good to add that here as well.

See https://github.com/apache/spark/blob/master/mllib/src/main/scala/org/apache/spark/mllib/optimization/GradientDescent.scala#L99