Documentation
¶
Overview ¶
Package svm implement svm classifier
Index ¶
- func AddData(d C.ISVMTrainer, labelID int, feats []float32)
- func Classify(d C.ISVMClassifier, vec []float32) ([]float64, error)
- func DestroyClassifier(d C.ISVMClassifier)
- func DestroyTrainer(d C.ISVMTrainer)
- func LoadClassifierModel(d C.ISVMClassifier, modelPath string)
- func Predict(d C.ISVMClassifier, vec []float32) float64
- func ResetTrainer(d C.ISVMTrainer)
- func SetFeatures(d C.ISVMTrainer, feats int)
- func SetLabels(d C.ISVMTrainer, labels int)
- func Train(d C.ISVMTrainer, modelPath string) error
- type BinaryClassifier
- type BinaryTrainer
- type Classifier
- type MultiClassClassifier
- type MultiClassTrainer
- type Trainer
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func AddData ¶
func AddData(d C.ISVMTrainer, labelID int, feats []float32)
AddData add data with label
func Classify ¶
func Classify(d C.ISVMClassifier, vec []float32) ([]float64, error)
Classify returns class scores
func DestroyClassifier ¶
func DestroyClassifier(d C.ISVMClassifier)
Destroy destroy C.ISVMClassifier
func LoadClassifierModel ¶
func LoadClassifierModel(d C.ISVMClassifier, modelPath string)
LoadModel load model
Types ¶
type BinaryClassifier ¶
type BinaryClassifier struct {
// contains filtered or unexported fields
}
BinaryClassifier represents svm classifier
func NewBinaryClassifier ¶
func NewBinaryClassifier() *BinaryClassifier
NewBinaryClassifier returns a new BinaryClassifier
func (*BinaryClassifier) Classify ¶
func (t *BinaryClassifier) Classify(vec []float32) ([]float64, error)
Classify returns classifid scores
func (*BinaryClassifier) Destroy ¶
func (t *BinaryClassifier) Destroy()
Destroy destroy C.ISVMClassifier
func (*BinaryClassifier) LoadModel ¶
func (t *BinaryClassifier) LoadModel(modelPath string)
LoadModel load model
func (*BinaryClassifier) Predict ¶
func (t *BinaryClassifier) Predict(vec []float32) float64
Predict returns predicted score
type BinaryTrainer ¶
type BinaryTrainer struct {
// contains filtered or unexported fields
}
BinaryTrainer represents svm trainer
func NewBinaryTrainer ¶
func NewBinaryTrainer() *BinaryTrainer
NewBinaryTrainer returns a new BinaryTrainer
func (*BinaryTrainer) AddData ¶
func (t *BinaryTrainer) AddData(labelID int, feats []float32)
AddData add data with label
func (*BinaryTrainer) Labels ¶
func (t *BinaryTrainer) Labels(labels int)
SetLabels set total labels
func (*BinaryTrainer) SetFeatures ¶
func (t *BinaryTrainer) SetFeatures(feats int)
SetFeatures set total features
func (*BinaryTrainer) Train ¶
func (t *BinaryTrainer) Train(modelPath string) error
Train train model
type Classifier ¶
type Classifier interface {
LoadModel(string)
Destroy()
Predict(vec []float32) float64
Classify(vec []float32) (scores []float64, err error)
}
Classifier represents svm classifier
type MultiClassClassifier ¶
type MultiClassClassifier struct {
// contains filtered or unexported fields
}
MultiClassClassifier represents svm classifier
func NewMultiClassClassifier ¶
func NewMultiClassClassifier() *MultiClassClassifier
NewMultiClassClassifier returns a new MultiClassClassifier
func (*MultiClassClassifier) Classify ¶
func (t *MultiClassClassifier) Classify(vec []float32) ([]float64, error)
Classify returns classifid scores
func (*MultiClassClassifier) Destroy ¶
func (t *MultiClassClassifier) Destroy()
Destroy destroy C.ISVMClassifier
func (*MultiClassClassifier) LoadModel ¶
func (t *MultiClassClassifier) LoadModel(modelPath string)
LoadModel load model
func (*MultiClassClassifier) Predict ¶
func (t *MultiClassClassifier) Predict(vec []float32) float64
Predict returns predicted score
type MultiClassTrainer ¶
type MultiClassTrainer struct {
// contains filtered or unexported fields
}
MultiClassTrainer represents svm trainer
func NewMultiClassTrainer ¶
func NewMultiClassTrainer() *MultiClassTrainer
NewMultiClassTrainer returns a new MultiClassTrainer
func (*MultiClassTrainer) AddData ¶
func (t *MultiClassTrainer) AddData(labelID int, feats []float32)
AddData add data with label
func (*MultiClassTrainer) Destroy ¶
func (t *MultiClassTrainer) Destroy()
Destroy destroy C.ISVMTrainer
func (*MultiClassTrainer) SetFeatures ¶
func (t *MultiClassTrainer) SetFeatures(feats int)
SetFeatures set total features
func (*MultiClassTrainer) SetLabels ¶
func (t *MultiClassTrainer) SetLabels(labels int)
SetLabels set total labels
func (*MultiClassTrainer) Train ¶
func (t *MultiClassTrainer) Train(modelPath string) error
Train train model
Source Files