nn

type CheckPoint struct {
	Weights [][][]float32
	Biases  [][]float32
}

type Data struct {
	X [][]float32
	Y [][]float32
}

type Operator interface {
	// Step through iterations of training process.
	Step(trainingData, validationData *Data) (*TrainingOutput, error)

	// Predict the model outcome.
	Predict(data *Data) (*PredictionOutput, error)

	// Save obtains model checkpoint.
	Save() (*CheckPoint, error)

	// Load re-initializes trainer with a checkpoint.
	Load(checkPoint *CheckPoint) error
}

type PredictionOutput struct {
	Accuracy         float32
	TruthArgmax      []int64
	PredictionArgmax []int64
}

type TrainingOutput struct {
	CrossEntropy       float32
	TrainingAccuracy   float32
	ValidationAccuracy float32
}