galaktaglare

type ActivationFunc func(float64) float64

type BatchNormalization struct {
	Gamma    float64
	Beta     float64
	Mean     []float64
	Variance []float64
	Epsilon  float64
	Input    []float64
	Output   []float64
}

type DataAnalysis struct{}

type DecisionTree struct {
	Root *TreeNode
}

type DenseLayer struct {
	InputSize            int
	OutputSize           int
	Weights              [][]float64
	Biases               []float64
	Activation           ActivationFunc
	Inputs               []float64
	WeightDecayL1        float64
	WeightDecayL2        float64
	DropoutRate          float64
	DropoutMask          []float64
	LearningRateSchedule LearningRateSchedule
	BatchNorm            *BatchNormalization
}

type Example struct {
	Features []float64
	Label    string
}

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

type LearningRateSchedule struct {
	InitialRate float64
	Decay       float64
}

type NeuralNetwork struct {
	Layers []*DenseLayer
}

type SpeechConfig struct {
	Lang string `toml:"lang"`
	Text string `toml:"text"`
	Reco string `toml:"reco"`
}

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

type TrainingMonitor struct {
	Epochs        int
	DisplayPeriod int
}

type TreeNode struct {
	SplitFeature   int
	SplitValue     float64
	PredictedClass string
	Left           *TreeNode
	Right          *TreeNode
}

type Variable struct {
	Value     float64
	Gradient  float64
	Children  []*Variable
	Operation string
}