type AdamOptimizer struct {
	LearningRate float64
	Beta1        float64
	Beta2        float64
	M            []*Matrix
	V            []*Matrix
	T            int
	Eps          float64
	Parameters   []*VariableNode
}

type AddNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type ColSliceNode struct {
	X          Node
	Start, End int
	Value      *Matrix
	Name       string
	// contains filtered or unexported fields
}

type ColSumNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type ConvNode struct {
	X        Node
	Kernel   Node
	Stride   int
	Value    *Matrix
	XPadding int
	YPadding int
	Name     string
	// contains filtered or unexported fields
}

type CrossEntropyLossNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type DivElementNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type DropoutNode struct {
	X     Node
	P     float64 //Keep probability
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type GradThresholdNode struct {
	X         Node
	Value     *Matrix
	Threshold float64
	Name      string
	// contains filtered or unexported fields
}

type HConcatNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type L2Normalizer struct {
	Dim    int     `json:"dim"`
	Groups [][]int `json:"groups"`
}

type LogNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type MSELossNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type Matrix struct {
	Data []float64 `json:"data"`
	Rows int       `json:"rows"`
	Cols int       `json:"cols"`
}

type MinMaxScaler struct {
	Min    []float64 `json:"min"`
	Max    []float64 `json:"max"`
	Groups [][]int   `json:"groups"`
	Dim    int       `json:"dim"`
}

type Model struct {
	Parameters    []*VariableNode `json:"parameters"`
	InputScalers  []Scaler        `json:"input_scalers"`
	TargetScalers []Scaler        `json:"target_scalers"`
}

type MultiElementNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type MultiNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

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

type Node interface {
	Backward(grad *Matrix)
	Forward() *Matrix
	Reset()
	Tag(name string) Node
}

type Normalizer interface {
	Normalize([][]float64) [][]float64
}

type Optimizer interface {
	Step(batchSize int)
	Reset()
}

type PoolNode struct {
	X      Node
	Width  int
	Height int
	Stride int
	Value  *Matrix
	Flags  []int
	Name   string
	// contains filtered or unexported fields
}

type ReLuNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type ReshapeNode struct {
	X     Node
	Rows  int
	Cols  int
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type RobustScaler struct {
	Median []float64 `json:"median"`
	IQR    []float64 `json:"IQR"`
	Groups [][]int   `json:"groups"`
	Dim    int       `json:"dim"`
}

type RowSliceNode struct {
	X          Node
	Start, End int
	Value      *Matrix
	Name       string
	// contains filtered or unexported fields
}

type RowSumNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type SGDOptimizer struct {
	LearningRate float64
	Momentum     float64
	Velocity     []*Matrix
	Parameters   []*VariableNode
}

type ScaleNode struct {
	X     Node
	Rate  float64
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type Scaler interface {
	Fit(data [][]float64)
	Transform(data [][]float64) [][]float64
}

type SigmoidNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type SoftmaxNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type SubNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type TanhNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type TransNode struct {
	X     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type VConcatNode struct {
	X     Node
	Y     Node
	Value *Matrix
	Name  string
	// contains filtered or unexported fields
}

type ValueThresholdNode struct {
	X        Node
	Value    *Matrix
	MinValue float64
	MaxValue float64
	Name     string
	// contains filtered or unexported fields
}

type VariableNode struct {
	Name     string  `json:"name"`
	Value    *Matrix `json:"value"`
	Gradient *Matrix `json:"-"`
	// contains filtered or unexported fields
}

type ZScoreScaler struct {
	Mean         []float64 `json:"mean"`
	StdDeviation []float64 `json:"stdDeviation"`
	Groups       [][]int   `json:"groups"`
	Dim          int       `json:"dim"`
}