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
Click to show internal directories.
Click to hide internal directories.