Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Config ¶
type Config struct { gd.MethodConfig StepSize mat.Float Beta1 mat.Float Beta2 mat.Float Epsilon mat.Float }
Config provides configuration settings for a RAdam optimizer.
func NewConfig ¶
NewConfig returns a new RAdam Config. It panics if beta1 or beta2 are not in the range [0.0, 1.0).
func NewDefaultConfig ¶
func NewDefaultConfig() Config
NewDefaultConfig returns a new Config with generically reasonable default values.
type RAdam ¶
type RAdam struct { Config RoMax mat.Float // The maximum length of the approximated SMA. TimeStep int }
RAdam implements the RAdam gradient descent optimization method.
func (*RAdam) Delta ¶
Delta returns the difference between the current params and where the method wants it to be.
Click to show internal directories.
Click to hide internal directories.