deepmind

type Activation interface {
	Activate(x *Node) (*Node, error)
	Name() string
}

type FC struct {
	// contains filtered or unexported fields
}

type FCOpts struct {
	InputSize  int
	OutputSize int
	// Sigmoid for example, see "active.go" for more activations.
	// Activation is optional,  default is Linear.
	Activation string

	//Gaussian(0.0, 0.08), see  "initializer.go" for more initializers.
	// Initializer is optional,  default is Uniform(-1,1).
	Initializer string

	// Probability of Dropout, it uses randomly zeroes out a *Tensor with a probability
	// drawn from a uniform distribution. Only float32 or float64 type supported.
	// Optional, default is zero, means
	Dropout float64
}

type GRU struct {
	// contains filtered or unexported fields
}

type GRUOpts struct {
	InputSize  int
	HiddenSize int
	// Sigmoid for example, see "active.go" for more activations.
	// Activation is optional,  default is Tanh.
	Activation string

	//Gaussian(0.0, 0.08), see  "initializer.go" for more initializers.
	// Initializer is optional,  default is Uniform(-1,1).
	InitWh string
	InitWr string
	InitWu string

	// Probability of Dropout, it uses randomly zeroes out a *Tensor with a probability
	// drawn from a uniform distribution. Only float32 or float64 type supported.
	// Optional, default is zero, means
	Dropout float64
}

type Initializer func(opts []float64) (InitWFn, error)

type JsonSaver struct {
	reflectx.Reflector
	// contains filtered or unexported fields
}

type LSTM struct {
	// contains filtered or unexported fields
}

type LSTMOpts struct {
	InputSize  int
	HiddenSize int
	// Sigmoid for example, see "active.go" for more activations.
	// Activation is optional,  default is Tanh.
	Activation string

	//Gaussian(0.0, 0.08), see  "initializer.go" for more initializers.
	// Initializer is optional,  default is Uniform(-1,1).
	InitWf string
	InitWi string
	InitWo string
	InitWc string

	// Probability of Dropout, it uses randomly zeroes out a *Tensor with a probability
	// drawn from a uniform distribution. Only float32 or float64 type supported.
	// Optional, default is zero, means
	Dropout float64
}

type Layer interface {
	Forward(x *Node, states States) (rv *Node, err error)
	Learnables() Nodes

	// If vs is nil, the initializer indicated in the Options will be used.
	Init(g *ExprGraph, dt tensor.Dtype, vs map[string][]float64) error

	// Get the name of the layer
	Name() string
	// Get the options of the layer
	Options() interface{}
}

type LayerOpts struct {
	Names []string
	Opts  map[string]interface{}
}

type LossFunc func(output, target *Node) (cost *Node, err error)

type Model struct {
	Layers []Layer

	//init data from saver
	InitData map[string][]float64
	// contains filtered or unexported fields
}

type RNN struct {
	// contains filtered or unexported fields
}

type RNNOpts struct {
	InputSize  int
	HiddenSize int
	// Sigmoid for example, see "active.go" for more activations.
	// Activation is optional,  default is Tanh.
	Activation string

	//Gaussian(0.0, 0.08), see  "initializer.go" for more initializers.
	// Initializer is optional,  default is Uniform(-1,1).
	Initializer string

	// Probability of Dropout, it uses randomly zeroes out a *Tensor with a probability
	// drawn from a uniform distribution. Only float32 or float64 type supported.
	// Optional, default is zero, means
	Dropout float64
}

type Saver interface {
	Load() (*Model, error)
	LoadGraph() (*Model, error)
	LoadData() (map[string][]float64, error)
	Save(m *Model) error
}

type States map[string]*Node