lib/rumale/optimizer/nadam.rb in rumale-0.17.0 vs lib/rumale/optimizer/nadam.rb in rumale-0.17.1
- old
+ new
@@ -6,14 +6,9 @@
module Rumale
# This module consists of the classes that implement optimizers adaptively tuning hyperparameters.
module Optimizer
# Nadam is a class that implements Nadam optimizer.
#
- # @example
- # optimizer = Rumale::Optimizer::Nadam.new(learning_rate: 0.01, decay1: 0.9, decay2: 0.999)
- # estimator = Rumale::LinearModel::LinearRegression.new(optimizer: optimizer, random_seed: 1)
- # estimator.fit(samples, values)
- #
# *Reference*
# - T. Dozat, "Incorporating Nesterov Momentum into Adam," Tech. Repo. Stanford University, 2015.
class Nadam
include Base::BaseEstimator
include Validation