Documentation ¶
Overview ¶
LSTM enriched with a PolicyGradient to enable Dynamic Skip Connections.
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Model ¶
type Model struct { PolicyGradient *stack.Model Lambda float64 WIn *nn.Param `type:"weights"` WInRec *nn.Param `type:"weights"` BIn *nn.Param `type:"biases"` WOut *nn.Param `type:"weights"` WOutRec *nn.Param `type:"weights"` BOut *nn.Param `type:"biases"` WFor *nn.Param `type:"weights"` WForRec *nn.Param `type:"weights"` BFor *nn.Param `type:"biases"` WCand *nn.Param `type:"weights"` WCandRec *nn.Param `type:"weights"` BCand *nn.Param `type:"biases"` }
Model contains the serializable parameters.
type Processor ¶
type Processor struct { nn.BaseProcessor PolicyGradient *stack.Processor States []*State // contains filtered or unexported fields }
func (*Processor) Forward ¶
Forward performs the forward step for each input and returns the result.
func (*Processor) PolicyGradientLogProbActions ¶
func (*Processor) SetInitialState ¶
Click to show internal directories.
Click to hide internal directories.