Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type TransformLogistic ¶
type TransformLogistic struct{}
func (*TransformLogistic) NOutputGroups ¶
func (t *TransformLogistic) NOutputGroups() int
func (*TransformLogistic) Name ¶
func (t *TransformLogistic) Name() string
func (*TransformLogistic) Transform ¶
func (t *TransformLogistic) Transform(rawPredictions []float32, outputPredictions []float32, startIndex int) error
func (*TransformLogistic) Type ¶
func (t *TransformLogistic) Type() TransformType
type TransformRaw ¶
type TransformRaw struct {
NumOutputGroups int
}
func (*TransformRaw) NOutputGroups ¶
func (t *TransformRaw) NOutputGroups() int
func (*TransformRaw) Name ¶
func (t *TransformRaw) Name() string
func (*TransformRaw) Transform ¶
func (t *TransformRaw) Transform(rawPredictions []float32, outputPredictions []float32, startIndex int) error
func (*TransformRaw) Type ¶
func (t *TransformRaw) Type() TransformType
type TransformSoftmax ¶
type TransformSoftmax struct {
NClasses int
}
func (*TransformSoftmax) NOutputGroups ¶
func (t *TransformSoftmax) NOutputGroups() int
func (*TransformSoftmax) Name ¶
func (t *TransformSoftmax) Name() string
func (*TransformSoftmax) Transform ¶
func (t *TransformSoftmax) Transform(rawPredictions []float32, outputPredictions []float32, startIndex int) error
func (*TransformSoftmax) Type ¶
func (t *TransformSoftmax) Type() TransformType
type TransformType ¶
type TransformType int
TransformType is enum for various transformation functions that could be applied to the raw model results.
const ( // Raw is a TransformType that do nothing Raw TransformType = 0 // Logistic is a TransformType that apply logistic function in order to obtain // positive class probabilities Logistic TransformType = 1 // Softmax is a TransformType to obtain multiclass probabilities Softmax TransformType = 2 )
func (TransformType) Name ¶
func (t TransformType) Name() string
Click to show internal directories.
Click to hide internal directories.