transforms

package
v0.0.0-...-2fe1034 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Apr 23, 2022 License: Apache-2.0 Imports: 4 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type Accuracy

type Accuracy struct {
	MetricBase
	// contains filtered or unexported fields
}

func NewAccuracy

func NewAccuracy() *Accuracy

func (*Accuracy) Call

func (m *Accuracy) Call(y []backends.Tensor, t []backends.Tensor) []backends.Tensor

func (*Accuracy) ClassName

func (m *Accuracy) ClassName() string

func (*Accuracy) Init

func (m *Accuracy) Init(self base.Object)

type Cost

type Cost interface {
	base.Object
	Call(y backends.Value, t backends.Value) backends.Value
	Bprop(y backends.Value, t backends.Value) backends.Value
}

func ToCost

func ToCost(v interface{}) Cost

type CostArg

type CostArg struct {
	base.ArgDefBase
}

func NewCostArg

func NewCostArg() *CostArg

func NewCostArgOpt

func NewCostArgOpt(defval Cost) *CostArg

func (*CostArg) Expand

func (a *CostArg) Expand(v interface{}) (interface{}, bool)

type CostBase

type CostBase struct {
	base.ObjectBase
}

func (*CostBase) Init

func (c *CostBase) Init(self base.Object)

type CrossEntropyBinary

type CrossEntropyBinary struct {
	CostBase
	// contains filtered or unexported fields
}

func NewCrossEntropyBinary

func NewCrossEntropyBinary(args ...interface{}) *CrossEntropyBinary

func (*CrossEntropyBinary) Bprop

func (*CrossEntropyBinary) Call

func (*CrossEntropyBinary) ClassName

func (c *CrossEntropyBinary) ClassName() string

func (*CrossEntropyBinary) Init

func (c *CrossEntropyBinary) Init(self base.Object, args base.Args)

type CrossEntropyMulti

type CrossEntropyMulti struct {
	CostBase
	// contains filtered or unexported fields
}

func NewCrossEntropyMulti

func NewCrossEntropyMulti(args ...interface{}) *CrossEntropyMulti

func (*CrossEntropyMulti) Bprop

func (*CrossEntropyMulti) Call

func (*CrossEntropyMulti) ClassName

func (c *CrossEntropyMulti) ClassName() string

func (*CrossEntropyMulti) Init

func (c *CrossEntropyMulti) Init(self base.Object, args base.Args)

type Explin

type Explin struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewExplin

func NewExplin(args ...interface{}) *Explin

func (*Explin) Bprop

func (t *Explin) Bprop(x backends.Tensor) backends.Value

func (*Explin) Call

func (t *Explin) Call(x backends.Tensor) backends.Value

func (*Explin) ClassName

func (t *Explin) ClassName() string

func (*Explin) Init

func (t *Explin) Init(self base.Object, args base.Args)

type Identity

type Identity struct {
	TransformBase
}

func NewIdentity

func NewIdentity(args ...interface{}) *Identity

func (*Identity) Bprop

func (t *Identity) Bprop(x backends.Tensor) backends.Value

func (*Identity) Call

func (t *Identity) Call(x backends.Tensor) backends.Value

func (*Identity) ClassName

func (t *Identity) ClassName() string

func (*Identity) Init

func (t *Identity) Init(self base.Object, args base.Args)

type LogLoss

type LogLoss struct {
	MetricBase
	// contains filtered or unexported fields
}

func NewLogLoss

func NewLogLoss() *LogLoss

func (*LogLoss) Call

func (m *LogLoss) Call(y []backends.Tensor, t []backends.Tensor) []backends.Tensor

func (*LogLoss) ClassName

func (m *LogLoss) ClassName() string

func (*LogLoss) Init

func (m *LogLoss) Init(self base.Object)

type Logistic

type Logistic struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewLogistic

func NewLogistic(args ...interface{}) *Logistic

func (*Logistic) Bprop

func (t *Logistic) Bprop(x backends.Tensor) backends.Value

func (*Logistic) Call

func (t *Logistic) Call(x backends.Tensor) backends.Value

func (*Logistic) ClassName

func (t *Logistic) ClassName() string

func (*Logistic) Init

func (t *Logistic) Init(self base.Object, args base.Args)

type MeanSquared

type MeanSquared struct {
	CostBase
}

func NewMeanSquared

func NewMeanSquared() *MeanSquared

func (*MeanSquared) Bprop

func (*MeanSquared) Call

func (*MeanSquared) ClassName

func (c *MeanSquared) ClassName() string

func (*MeanSquared) Init

func (c *MeanSquared) Init(self base.Object)

type Metric

type Metric interface {
	base.Object
	MetricNames() []string
	Call(y []backends.Tensor, t []backends.Tensor) []backends.Tensor
}

type MetricBase

type MetricBase struct {
	base.ObjectBase
	// contains filtered or unexported fields
}

func (*MetricBase) Init

func (m *MetricBase) Init(self base.Object)

func (*MetricBase) MetricNames

func (m *MetricBase) MetricNames() []string

type Misclassification

type Misclassification struct {
	MetricBase
	// contains filtered or unexported fields
}

func NewMisclassification

func NewMisclassification(steps int) *Misclassification

func (*Misclassification) Call

func (*Misclassification) ClassName

func (m *Misclassification) ClassName() string

func (*Misclassification) Init

func (m *Misclassification) Init(self base.Object, steps int)

type MultiMetric

type MultiMetric struct {
	MetricBase
	// contains filtered or unexported fields
}

func NewMuitiMetric

func NewMuitiMetric(metric Metric, index int) *MultiMetric

func (*MultiMetric) Call

func (*MultiMetric) ClassName

func (m *MultiMetric) ClassName() string

func (*MultiMetric) Init

func (m *MultiMetric) Init(self base.Object, metric Metric, index int)

type Normalizer

type Normalizer struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewNormalizer

func NewNormalizer(args ...interface{}) *Normalizer

func (*Normalizer) Bprop

func (t *Normalizer) Bprop(x backends.Tensor) backends.Value

func (*Normalizer) Call

func (*Normalizer) ClassName

func (t *Normalizer) ClassName() string

func (*Normalizer) Init

func (t *Normalizer) Init(self base.Object, args base.Args)

type PixelwiseSoftmax

type PixelwiseSoftmax struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewPixelwiseSoftmax

func NewPixelwiseSoftmax(args ...interface{}) *PixelwiseSoftmax

func (*PixelwiseSoftmax) Bprop

func (*PixelwiseSoftmax) Call

func (*PixelwiseSoftmax) ClassName

func (t *PixelwiseSoftmax) ClassName() string

func (*PixelwiseSoftmax) Init

func (t *PixelwiseSoftmax) Init(self base.Object, args base.Args)

type PrecisionRecall

type PrecisionRecall struct {
	MetricBase
	// contains filtered or unexported fields
}

func NewPrecisionRecall

func NewPrecisionRecall(numClasses int, binarize bool, epsilon float64) *PrecisionRecall

func (*PrecisionRecall) Call

func (*PrecisionRecall) ClassName

func (m *PrecisionRecall) ClassName() string

func (*PrecisionRecall) Init

func (m *PrecisionRecall) Init(self base.Object, numClasses int, binarize bool, epsilon float64)

type Rectlin

type Rectlin struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewRectlin

func NewRectlin(args ...interface{}) *Rectlin

func (*Rectlin) Bprop

func (t *Rectlin) Bprop(x backends.Tensor) backends.Value

func (*Rectlin) Call

func (t *Rectlin) Call(x backends.Tensor) backends.Value

func (*Rectlin) ClassName

func (t *Rectlin) ClassName() string

func (*Rectlin) Init

func (t *Rectlin) Init(self base.Object, args base.Args)

type Rectlinclip

type Rectlinclip struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewRectlinclip

func NewRectlinclip(args ...interface{}) *Rectlinclip

func (*Rectlinclip) Bprop

func (*Rectlinclip) Call

func (*Rectlinclip) ClassName

func (t *Rectlinclip) ClassName() string

func (*Rectlinclip) Init

func (t *Rectlinclip) Init(self base.Object, args base.Args)

type Sign

type Sign struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewSign

func NewSign(args ...interface{}) *Sign

func (*Sign) Bprop

func (t *Sign) Bprop(x backends.Tensor) backends.Value

func (*Sign) Call

func (t *Sign) Call(x backends.Tensor) backends.Value

func (*Sign) ClassName

func (t *Sign) ClassName() string

func (*Sign) Init

func (t *Sign) Init(self base.Object, args base.Args)

type SmoothL1Loss

type SmoothL1Loss struct {
	CostBase
	// contains filtered or unexported fields
}

func NewSmoothL1Loss

func NewSmoothL1Loss(args ...interface{}) *SmoothL1Loss

func (*SmoothL1Loss) Bprop

func (*SmoothL1Loss) Call

func (*SmoothL1Loss) ClassName

func (c *SmoothL1Loss) ClassName() string

func (*SmoothL1Loss) Init

func (c *SmoothL1Loss) Init(self base.Object, args base.Args)

type Softmax

type Softmax struct {
	TransformBase
	// contains filtered or unexported fields
}

func NewSoftmax

func NewSoftmax(args ...interface{}) *Softmax

func (*Softmax) Bprop

func (t *Softmax) Bprop(x backends.Tensor) backends.Value

func (*Softmax) Call

func (t *Softmax) Call(x backends.Tensor) backends.Value

func (*Softmax) ClassName

func (t *Softmax) ClassName() string

func (*Softmax) Init

func (t *Softmax) Init(self base.Object, args base.Args)

type SquareHingeLoss

type SquareHingeLoss struct {
	CostBase
	// contains filtered or unexported fields
}

func NewSquareHingeLoss

func NewSquareHingeLoss(args ...interface{}) *SquareHingeLoss

func (*SquareHingeLoss) Bprop

func (*SquareHingeLoss) Call

func (*SquareHingeLoss) ClassName

func (c *SquareHingeLoss) ClassName() string

func (*SquareHingeLoss) Init

func (c *SquareHingeLoss) Init(self base.Object, args base.Args)

type SumSquared

type SumSquared struct {
	CostBase
}

func NewSumSquared

func NewSumSquared() *SumSquared

func (*SumSquared) Bprop

func (*SumSquared) Call

func (*SumSquared) ClassName

func (c *SumSquared) ClassName() string

func (*SumSquared) Init

func (c *SumSquared) Init(self base.Object)

type Tanh

type Tanh struct {
	TransformBase
}

func NewTanh

func NewTanh(args ...interface{}) *Tanh

func (*Tanh) Bprop

func (t *Tanh) Bprop(x backends.Tensor) backends.Value

func (*Tanh) Call

func (t *Tanh) Call(x backends.Tensor) backends.Value

func (*Tanh) ClassName

func (t *Tanh) ClassName() string

func (*Tanh) Init

func (t *Tanh) Init(self base.Object, args base.Args)

type TopKMisclassification

type TopKMisclassification struct {
	MetricBase
	// contains filtered or unexported fields
}

func NewTopKMisclassification

func NewTopKMisclassification(k int) *TopKMisclassification

func (*TopKMisclassification) Call

func (*TopKMisclassification) ClassName

func (m *TopKMisclassification) ClassName() string

func (*TopKMisclassification) Init

func (m *TopKMisclassification) Init(self base.Object, k int)

type Transform

type Transform interface {
	base.Object
	Call(x backends.Tensor) backends.Value
	Bprop(x backends.Tensor) backends.Value
}

func ToTransform

func ToTransform(v interface{}) Transform

type TransformArg

type TransformArg struct {
	base.ArgDefBase
}

func NewTransformArg

func NewTransformArg() *TransformArg

func NewTransformArgOpt

func NewTransformArgOpt(defval Transform) *TransformArg

func (*TransformArg) Expand

func (a *TransformArg) Expand(v interface{}) (interface{}, bool)

type TransformBase

type TransformBase struct {
	base.ObjectBase
	// contains filtered or unexported fields
}

func (*TransformBase) Init

func (t *TransformBase) Init(self base.Object, args base.Args)

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL