lib/rumale/optimizer/rmsprop.rb in rumale-0.13.8 vs lib/rumale/optimizer/rmsprop.rb in rumale-0.14.0

@@ -23,10 +23,10 @@
       #
       # @param learning_rate [Float] The initial value of learning rate.
       # @param momentum [Float] The initial value of momentum.
       # @param decay [Float] The smooting parameter.
       def initialize(learning_rate: 0.01, momentum: 0.9, decay: 0.9)
-        check_params_float(learning_rate: learning_rate, momentum: momentum, decay: decay)
+        check_params_numeric(learning_rate: learning_rate, momentum: momentum, decay: decay)
         check_params_positive(learning_rate: learning_rate, momentum: momentum, decay: decay)
         @params = {}
         @params[:learning_rate] = learning_rate
         @params[:momentum] = momentum
         @params[:decay] = decay