Documentation
¶
Index ¶
- func AlgorithmRun(ctx *cli.Context) (err error)
- func AlgorithmRunOnDataSet(classifier Classifier, trainSet, testSet *core.DataSet) (float64, []*eval.LabelPrediction)
- func Commands() []cli.Command
- func MultiClassRun(ctx *cli.Context) (err error)
- func MultiClassRunOnDataSet(classifier MultiClassClassifier, trainSet, testSet *core.DataSet) (accuracy float64, predictLabels []int)
- type Classifier
- type MultiClassClassifier
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func AlgorithmRun ¶
func AlgorithmRunOnDataSet ¶
func AlgorithmRunOnDataSet(classifier Classifier, trainSet, testSet *core.DataSet) (float64, []*eval.LabelPrediction)
func MultiClassRun ¶
func MultiClassRunOnDataSet ¶
func MultiClassRunOnDataSet(classifier MultiClassClassifier, trainSet, testSet *core.DataSet) (accuracy float64, predictLabels []int)
Types ¶
type Classifier ¶
type Classifier interface {
internal.Algorithm
//Train model on a given dataset
Train(dataset *core.DataSet)
//Predict the probability of a sample to be positive sample
Predict(sample *core.Sample) float64
SaveModel(path string)
LoadModel(path string)
}
func GetClassifier ¶
func GetClassifier(method string) Classifier
type MultiClassClassifier ¶
type MultiClassClassifier interface {
internal.Algorithm
//Train model on a given dataset
Train(dataset *core.DataSet)
//Predict the probability of a sample to be positive sample
PredictMultiClass(sample *core.Sample) *core.ArrayVector
SaveModel(path string)
LoadModel(path string)
}
func GetMutliClassClassifier ¶
func GetMutliClassClassifier(method string) MultiClassClassifier
Source Files
Directories
Click to show internal directories.
Click to hide internal directories.