Module mls_lib.objects.optimizers.adams_optimizer

AdamsOptimizer: Component that trains and makes predictions.

Classes

class AdamsOptimizer (learning_rate, beta1, beta2, epsilon)

AdamsOptimizer: Component that trains and makes predictions.

Expand source code

class AdamsOptimizer(Object):
    """ AdamsOptimizer: Component that trains and makes predictions. """
    def __init__(self, learning_rate, beta1, beta2, epsilon) -> None:
        super().__init__()
        self.learning_rate = learning_rate
        self.beta1 = beta1
        self.beta2 = beta2
        self.epsilon = epsilon
    def get_params(self):
        """ Returns the parameters of the optimizer. """
        return {
            "learning_rate": self.learning_rate,
            "beta1": self.beta1,
            "beta2": self.beta2,
            "epsilon": self.epsilon
        }

Ancestors

• Object

Methods

def get_params(self)

Returns the parameters of the optimizer.

Inherited members

• Object:
  • copy