Versions in this module Expand all Collapse all v0 v0.1.1 May 13, 2023 Changes in this version + var DefaultAgentConfig = &AgentConfig + var DefaultAtariLayerBuilder = func(x, y *modelv1.Input) []layer.Config + var DefaultAtariPolicyConfig = &PolicyConfig + var DefaultFCLayerBuilder = func(x, y *modelv1.Input) []layer.Config + var DefaultHyperparameters = &Hyperparameters + var DefaultPolicyConfig = &PolicyConfig + func MakePolicy(name string, config *PolicyConfig, base *agentv1.Base, env *envv1.Env) (modelv1.Model, error) + type Agent struct + Epsilon common.Schedule + Policy model.Model + TargetPolicy model.Model + func NewAgent(c *AgentConfig, env *envv1.Env) (*Agent, error) + func (a *Agent) Action(state *tensor.Dense) (action int, err error) + func (a *Agent) Learn() error + func (a *Agent) Remember(event *Event) + type AgentConfig struct + Base *agentv1.Base + PolicyConfig *PolicyConfig + type Event struct + Action int + State *tensor.Dense + func NewEvent(state *tensor.Dense, action int, outcome *envv1.Outcome) *Event + func (e *Event) Print() + type Hyperparameters struct + BufferSize int + Epsilon common.Schedule + Gamma float32 + UpdateTargetSteps int + type LayerBuilder func(x, y *modelv1.Input) []layer.Config + type Memory struct + func NewMemory() *Memory + func (m *Memory) Sample(batchsize int) ([]*Event, error) + type PolicyConfig struct + BatchSize int + LayerBuilder LayerBuilder + Loss modelv1.Loss + Optimizer g.Solver + Track bool
Go to main page