goras

Jan 1, 2024

Changes in this version

+ func MustGetTensorDataType(t interface{}) tensor.Dtype

+ func MustMake1DSliceTensor[T any](data []T) tensor.Tensor

+ func MustMake2DSliceTensor[T any](data [][]T) tensor.Tensor

type Model

+ func (m *Model) MustBindParamsFrom(m1 *Model)

+ func (m *Model) MustCopyParamsFrom(m1 *Model)

+ func (m *Model) MustFit(xs, ys map[string]T.Tensor, solver G.Solver, opts ...FitOpt)

+ func (m *Model) MustFitBatch(inputs, lossRequirements map[string]T.Tensor, solver G.Solver) float64

+ func (m *Model) MustFitGenerator(tdg TrainingDataGenerator, solver G.Solver, opts ...FitOpt)

+ func (m *Model) MustPredict(xs map[string]T.Tensor) map[string]T.Tensor

+ func (m *Model) MustPredictBatch(inputs map[string]T.Tensor) map[string]T.Tensor

+ func (m *Model) MustReadParams(r io.Reader)

+ func (m *Model) MustSetParams(params map[string]*T.Dense)

+ func (m *Model) MustWriteParams(w io.Writer)

Dec 29, 2023

Changes in this version

+ func Make1DSliceTensor[T any](data []T) (tensor.Tensor, error)

+ func Make2DSliceTensor[T any](data [][]T) (tensor.Tensor, error)

+ type OneHotLayer struct

+ DType T.Dtype

+ NumClasses int

+ func OneHot(m *Model, name string, dtype T.Dtype, numClasses int) *OneHotLayer

+ func (*OneHotLayer) Parameters() map[string]*G.Node

+ func (l *OneHotLayer) Attach(n *G.Node) (*G.Node, error)

+ func (l *OneHotLayer) MustAttach(n *G.Node) *G.Node

Dec 26, 2023

Changes in this version

+ func Conv1DSliceToTensor[T any](data []T) (tensor.Tensor, error)

+ func Conv2DSliceToTensor[T any](data [][]T) (tensor.Tensor, error)

+ func GetTensorDataType(t interface{}) (tensor.Dtype, error)

type BuildOpts

+ func WithInput(inputName string, inputNode *G.Node) BuildOpts

+ func WithLoss(loss LossFunc) BuildOpts

+ func WithOutput(name string, outputNode *G.Node) BuildOpts

+ type LossFunc func() (lossOut *G.Node, lossInps map[string]*G.Node, err error)

+ func BCELoss(targetName string, output *G.Node) LossFunc

+ func CCELoss(targetName string, output *G.Node) LossFunc

+ func L2Loss(layers ...Layer) LossFunc

+ func MSELoss(targetName string, output *G.Node) LossFunc

+ func WeightedAdditiveLoss(losses []LossFunc, weights []float64) LossFunc

type Model

+ LossRequiredNodes map[string]*G.Node

+ OutputNodes map[string]*G.Node

+ OutputValues map[string]*G.Value

+ func (m *Model) CopyParamsFrom(m1 *Model) error

+ type NamedTs map[string]T.Tensor

Dec 3, 2023

Changes in this version

+ var ImageUtils imageUtils = imageUtils

type Model

+ func (m *Model) FitGenerator(tdg TrainingDataGenerator, solver G.Solver, opts ...FitOpt) error

+ type TensorTrainingDataGenerator struct

+ func NewTTDG(xs, ys []T.Tensor) *TensorTrainingDataGenerator

+ func (t *TensorTrainingDataGenerator) NextBatch(int) ([]T.Tensor, []T.Tensor, error)

+ func (t *TensorTrainingDataGenerator) NumBatches() int

+ func (t *TensorTrainingDataGenerator) Reset(batchSize int) error

+ type TrainingDataGenerator interface

+ NextBatch func(batchSize int) ([]T.Tensor, []T.Tensor, error)

+ NumBatches func() int

+ Reset func(batchSize int) error

Aug 11, 2023

Changes in this version

+ type EpochCallback func(epoch int) error

+ func RepeatedSaveModelParametersCallback(model *Model, pathWithFormat string, every int) EpochCallback

+ func SaveModelParametersCallback(model *Model, path string) EpochCallback

type FitOpt

+ func WithEpochCallback(cb EpochCallback) FitOpt

type Layer

+ INodes func() []*G.Node

+ Node func() *G.Node

type LayerBase

+ InputNodes []*G.Node

+ LayerType string

+ OutputNode *G.Node

+ func (l *LayerBase) INodes() []*G.Node

+ func (l *LayerBase) Node() *G.Node

+ func (l *LayerBase) Type() string

type Model

+ func (m *Model) Predict(xs []T.Tensor) ([]T.Tensor, error)

+ func (m *Model) Summary() string

type ReshapeLayer

+ func (l *ReshapeLayer) Parameters() map[string]*G.Node

Aug 10, 2023

Changes in this version

+ func V(ts ...T.Tensor) []T.Tensor

type ActivationLayer

+ LeakyReluGrad float64

+ func Binary(m *Model, name string) *ActivationLayer

+ func LeakyRelu(m *Model, name string, grad ...float64) *ActivationLayer

+ func Relu(m *Model, name string) *ActivationLayer

+ func Sigmoid(m *Model, name string) *ActivationLayer

+ func Softmax(m *Model, name string) *ActivationLayer

+ func Tanh(m *Model, name string) *ActivationLayer

+ func (l *ActivationLayer) Type() string

+ type BuildOpts func(*buildParams)

+ func WithInputs(inputNodes ...*G.Node) BuildOpts

+ func WithLosses(losses ...func(*G.Node, *G.Node) (*G.Node, error)) BuildOpts

+ func WithOutputs(outputNodes ...*G.Node) BuildOpts

+ type Conv2DLayer struct

+ KernelSize []int

+ Kernels *G.Node

+ NumKernels int

+ Padding string

+ Stride []int

+ func Conv2D(m *Model, name string, kernelShape, stride []int, padding string, ...) *Conv2DLayer

+ func SimpleConv2D(m *Model, name string, kernelSize int, numKernels int) *Conv2DLayer

+ func (l *Conv2DLayer) Attach(x *G.Node) (*G.Node, error)

+ func (l *Conv2DLayer) MustAttach(n *G.Node) *G.Node

+ func (l *Conv2DLayer) Parameters() map[string]*G.Node

+ func (l *Conv2DLayer) Type() string

type DenseLayer

+ func (l *DenseLayer) Type() string

type DropoutLayer

+ func (d *DropoutLayer) Type() string

type InputLayer

+ func (l *InputLayer) Type() string

type Layer

+ Type func() string

type LayerBase

+ DType T.Dtype

+ type MaxPooling2DLayer struct

+ Padding string

+ PoolSize []int

+ Stride []int

+ func MaxPooling2D(m *Model, name string, poolSize, stride []int, padding string) *MaxPooling2DLayer

+ func SimpleMaxPooling2D(m *Model, name string, poolSize int) *MaxPooling2DLayer

+ func (l *MaxPooling2DLayer) Attach(x *G.Node) (*G.Node, error)

+ func (l *MaxPooling2DLayer) MustAttach(x *G.Node) *G.Node

+ func (l *MaxPooling2DLayer) Parameters() map[string]*G.Node

+ func (l *MaxPooling2DLayer) Type() string

type Model

+ DType T.Dtype

+ InputNodes []*G.Node

+ type ReshapeLayer struct

+ ToShape T.Shape

+ func Reshape(model *Model, name string, newShape T.Shape) *ReshapeLayer

+ func (l *ReshapeLayer) Attach(n *G.Node) (*G.Node, error)

+ func (l *ReshapeLayer) MustAttach(n *G.Node) *G.Node

+ func (l *ReshapeLayer) Type() string

Aug 9, 2023

Changes in this version

type ActivationLayer

+ func (a *ActivationLayer) MustAttach(n *G.Node) *G.Node

type DenseLayer

+ func (d *DenseLayer) MustAttach(n *G.Node) *G.Node

type DropoutLayer

+ func (d *DropoutLayer) MustAttach(n *G.Node) *G.Node

+ type FitOpt func(*fitParams)

+ func WithClearLine(clear bool) FitOpt

+ func WithEpochs(epochs int) FitOpt

+ func WithLoggingEvery(epochs int) FitOpt

+ func WithVerbose(verbose bool) FitOpt

type Model

+ func (m *Model) Fit(x, y *T.Dense, solver G.Solver, opts ...FitOpt) error

+ func (m *Model) MustBuild(inputNode, outputNode *G.Node, loss func(*G.Node, *G.Node) (*G.Node, error))

Aug 9, 2023

Changes in this version

+ func BCE(output, target *G.Node) *G.Node

+ func CCE(output, target *G.Node) *G.Node

+ func IsValidActivation(ac string) bool

type Layer

+ Trainable func() bool

type LayerBase

+ IsTrainable bool

+ func (l *LayerBase) Trainable() bool

Aug 9, 2023

Changes in this version

+ func MSE(output, target *G.Node) *G.Node

+ type ActivationLayer struct

+ Activation string

+ func Activation(m *Model, name string, activation string) *ActivationLayer

+ func (a *ActivationLayer) Attach(n *G.Node) *G.Node

+ func (a *ActivationLayer) Parameters() map[string]*G.Node

+ type DenseLayer struct

+ Nodes int

+ Weights *G.Node

+ func Dense(m *Model, name string, nodes int) *DenseLayer

+ func (d *DenseLayer) Attach(n *G.Node) *G.Node

+ func (d *DenseLayer) Parameters() map[string]*G.Node

+ type DropoutLayer struct

+ DropoutProbability float64

+ func Dropout(m *Model, name string, dropoutProbability float64) *DropoutLayer

+ func (d *DropoutLayer) Attach(n *G.Node) *G.Node

+ func (d *DropoutLayer) Parameters() map[string]*G.Node

+ type InputLayer struct

+ Node *G.Node

+ func Input(m *Model, name string, shape ...int) *InputLayer

+ func (i *InputLayer) Parameters() map[string]*G.Node

+ type Layer interface

+ Name func() string

+ Parameters func() map[string]*G.Node

+ type LayerBase struct

+ Graph *G.ExprGraph

+ LayerName string

+ func (l *LayerBase) Name() string

+ type Model struct

+ Graph *G.ExprGraph

+ InputNode *G.Node

+ Layers []Layer

+ LossValue G.Value

+ Machine G.VM

+ OutputNode *G.Node

+ OutputValue G.Value

+ TargetOutputNode *G.Node

+ func NewModel() *Model

+ func (m *Model) AddLayer(l Layer)

+ func (m *Model) BindParamsFrom(m1 *Model)

+ func (m *Model) Build(inputNode, outputNode *G.Node, loss func(*G.Node, *G.Node) *G.Node)

+ func (m *Model) FitBatch(input, target *T.Dense, solver G.Solver) float64

+ func (m *Model) GetParams() map[string]*T.Dense

+ func (m *Model) PredictBatch(input *T.Dense) *T.Dense

+ func (m *Model) ReadParams(r io.Reader)

+ func (m *Model) SetParams(params map[string]*T.Dense)

+ func (m *Model) Trainables() G.Nodes

+ func (m *Model) WriteParams(w io.Writer)

+ type Namer struct

+ BaseName string

+ Counter int

+ func NewNamer(baseName string) *Namer

+ func (n *Namer) Next() string