Documentation ¶
Index ¶
- type Accuracy
- type Cost
- type CostArg
- type CostBase
- type CrossEntropyBinary
- type CrossEntropyMulti
- type Explin
- type Identity
- type LogLoss
- type Logistic
- type MeanSquared
- type Metric
- type MetricBase
- type Misclassification
- type MultiMetric
- type Normalizer
- type PixelwiseSoftmax
- type PrecisionRecall
- type Rectlin
- type Rectlinclip
- type Sign
- type SmoothL1Loss
- type Softmax
- type SquareHingeLoss
- type SumSquared
- type Tanh
- type TopKMisclassification
- type Transform
- type TransformArg
- type TransformBase
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
type Cost ¶
type CostArg ¶
type CostArg struct {
base.ArgDefBase
}
func NewCostArg ¶
func NewCostArg() *CostArg
func NewCostArgOpt ¶
type CostBase ¶
type CostBase struct {
base.ObjectBase
}
type CrossEntropyBinary ¶
type CrossEntropyBinary struct { CostBase // contains filtered or unexported fields }
func NewCrossEntropyBinary ¶
func NewCrossEntropyBinary(args ...interface{}) *CrossEntropyBinary
func (*CrossEntropyBinary) ClassName ¶
func (c *CrossEntropyBinary) ClassName() string
type CrossEntropyMulti ¶
type CrossEntropyMulti struct { CostBase // contains filtered or unexported fields }
func NewCrossEntropyMulti ¶
func NewCrossEntropyMulti(args ...interface{}) *CrossEntropyMulti
func (*CrossEntropyMulti) ClassName ¶
func (c *CrossEntropyMulti) ClassName() string
type Explin ¶
type Explin struct { TransformBase // contains filtered or unexported fields }
type Identity ¶
type Identity struct {
TransformBase
}
func NewIdentity ¶
func NewIdentity(args ...interface{}) *Identity
type LogLoss ¶
type LogLoss struct { MetricBase // contains filtered or unexported fields }
func NewLogLoss ¶
func NewLogLoss() *LogLoss
type Logistic ¶
type Logistic struct { TransformBase // contains filtered or unexported fields }
func NewLogistic ¶
func NewLogistic(args ...interface{}) *Logistic
type MeanSquared ¶
type MeanSquared struct {
CostBase
}
func NewMeanSquared ¶
func NewMeanSquared() *MeanSquared
func (*MeanSquared) ClassName ¶
func (c *MeanSquared) ClassName() string
func (*MeanSquared) Init ¶
func (c *MeanSquared) Init(self base.Object)
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) ClassName ¶
func (m *Misclassification) ClassName() string
type MultiMetric ¶
type MultiMetric struct { MetricBase // contains filtered or unexported fields }
func NewMuitiMetric ¶
func NewMuitiMetric(metric Metric, index int) *MultiMetric
func (*MultiMetric) ClassName ¶
func (m *MultiMetric) ClassName() string
type Normalizer ¶
type Normalizer struct { TransformBase // contains filtered or unexported fields }
func NewNormalizer ¶
func NewNormalizer(args ...interface{}) *Normalizer
func (*Normalizer) ClassName ¶
func (t *Normalizer) ClassName() string
type PixelwiseSoftmax ¶
type PixelwiseSoftmax struct { TransformBase // contains filtered or unexported fields }
func NewPixelwiseSoftmax ¶
func NewPixelwiseSoftmax(args ...interface{}) *PixelwiseSoftmax
func (*PixelwiseSoftmax) ClassName ¶
func (t *PixelwiseSoftmax) ClassName() string
type PrecisionRecall ¶
type PrecisionRecall struct { MetricBase // contains filtered or unexported fields }
func NewPrecisionRecall ¶
func NewPrecisionRecall(numClasses int, binarize bool, epsilon float64) *PrecisionRecall
func (*PrecisionRecall) ClassName ¶
func (m *PrecisionRecall) ClassName() string
type Rectlin ¶
type Rectlin struct { TransformBase // contains filtered or unexported fields }
func NewRectlin ¶
func NewRectlin(args ...interface{}) *Rectlin
type Rectlinclip ¶
type Rectlinclip struct { TransformBase // contains filtered or unexported fields }
func NewRectlinclip ¶
func NewRectlinclip(args ...interface{}) *Rectlinclip
func (*Rectlinclip) ClassName ¶
func (t *Rectlinclip) ClassName() string
type Sign ¶
type Sign struct { TransformBase // contains filtered or unexported fields }
type SmoothL1Loss ¶
type SmoothL1Loss struct { CostBase // contains filtered or unexported fields }
func NewSmoothL1Loss ¶
func NewSmoothL1Loss(args ...interface{}) *SmoothL1Loss
func (*SmoothL1Loss) ClassName ¶
func (c *SmoothL1Loss) ClassName() string
type Softmax ¶
type Softmax struct { TransformBase // contains filtered or unexported fields }
func NewSoftmax ¶
func NewSoftmax(args ...interface{}) *Softmax
type SquareHingeLoss ¶
type SquareHingeLoss struct { CostBase // contains filtered or unexported fields }
func NewSquareHingeLoss ¶
func NewSquareHingeLoss(args ...interface{}) *SquareHingeLoss
func (*SquareHingeLoss) ClassName ¶
func (c *SquareHingeLoss) ClassName() string
type SumSquared ¶
type SumSquared struct {
CostBase
}
func NewSumSquared ¶
func NewSumSquared() *SumSquared
func (*SumSquared) ClassName ¶
func (c *SumSquared) ClassName() string
func (*SumSquared) Init ¶
func (c *SumSquared) Init(self base.Object)
type Tanh ¶
type Tanh struct {
TransformBase
}
type TopKMisclassification ¶
type TopKMisclassification struct { MetricBase // contains filtered or unexported fields }
func NewTopKMisclassification ¶
func NewTopKMisclassification(k int) *TopKMisclassification
func (*TopKMisclassification) ClassName ¶
func (m *TopKMisclassification) ClassName() string
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 }
Click to show internal directories.
Click to hide internal directories.