package
0.0.0-20210130030158-f83957ed3c81
Repository: https://github.com/aunum/goro.git
Documentation: pkg.go.dev
# Functions
AsBatch adds a batch size to a clone opt.
AsType explicitly sets the type of the input.
NameAsBatch takes an input name and converts it to its batch name.
NewInput returns a new input.
NewOpts returns a new set of options for a model.
NewPseudoHuberLoss return a new huber loss.
NewSequential returns a new sequential model.
ValuesFrom returns the value as an array of gorgonia values.
WithBatchSize sets the batch size for the model.
WithGraphLogger adds a logger to the model which will print out the graph operations as they occur.
WithLogger adds a logger to the model.
WithLoss uses a specific loss function with the model.
WithMetrics sets the metrics that the model should track.
WithOptimizer uses a specific optimizer function.
WithoutTracker uses no tracking with the model.
WithTracker adds a tracker to the model, if not provided one will be created.
# Constants
TrainBatchLossMetric is the metric for batch training loss.
TrainLossMetric is the metric for training loss.
# Variables
AllMetrics are all metrics.
CrossEntropy loss.
MSE is standard mean squared error loss.
PseudoCrossEntropy loss.
PseudoHuber is the Huber loss function.
# Structs
CrossEntropyLoss is standard cross entropy loss.
Input into the model.
InputLayer is an input layer to be used in a chain.
MSELoss is mean squared error loss.
Opts are optsion for a model.
PseudoCrossEntropyLoss is standard cross entropy loss.
PseudoHuberLoss is a loss that is less sensetive to outliers.
Sequential model.