lib/rumale/optimizer/rmsprop.rb in rumale-0.17.0 vs lib/rumale/optimizer/rmsprop.rb in rumale-0.17.1
- old
+ new
@@ -5,14 +5,9 @@
module Rumale
module Optimizer
# RMSProp is a class that implements RMSProp optimizer.
#
- # @example
- # optimizer = Rumale::Optimizer::RMSProp.new(learning_rate: 0.01, momentum: 0.9, decay: 0.9)
- # estimator = Rumale::LinearModel::LinearRegression.new(optimizer: optimizer, random_seed: 1)
- # estimator.fit(samples, values)
- #
# *Reference*
# - I. Sutskever, J. Martens, G. Dahl, and G. Hinton, "On the importance of initialization and momentum in deep learning," Proc. ICML' 13, pp. 1139--1147, 2013.
# - G. Hinton, N. Srivastava, and K. Swersky, "Lecture 6e rmsprop," Neural Networks for Machine Learning, 2012.
class RMSProp
include Base::BaseEstimator