Skip to content

VinceShieh/Spark-AdaOptimizer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Spark-AdaOptimizer

A Spark-based implementation of Adam and AdaGrad optimizer, methods for Stochastic Optimization. See https://arxiv.org/abs/1412.6980 and http://www.jmlr.org/papers/volume12/duchi11a/duchi11a.pdf Comparing with SGD, Adam and AdaGrad have better performance. Especially in case of sparse features, Adam can converge faster than normal SGD.

How to try

After getting Spark build environment ready, run 'sbt package' under the root of the source to build the package then excute '$SPARKHOME/bin/spark-submit --class TestAdaOptimizer --master local[*] target/scala-2.10/spark-adaoptimizer_2.10-0.0.1.jar' ##FAQ "Some APIs are not found or parameters not match"

Please modify the settings in build.sbt to match your environments.

Contact & Feedback

If you encounter bugs, feel free to submit an issue or pull request.

About

Spark-based implementation of Adagrad and Adam solver

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages