clf

Jan 28, 2023

Changes in this version

+ func GetConfusionMatrix(classifConfig *Config) (confusionMatrix [][]int, yTrue, yPred []int)

+ func GetSums(confusionMatrix [][]int) (sumI, sumJ []int, sumIJ, sumTrue int)

+ type Scores struct

+ Accuracy float64

+ F1score []float64

+ F1scoreAvg float64

+ F1scoreWeightedAvg float64

+ Precision []float64

+ PrecisionAvg float64

+ PrecisionWeightedAvg float64

+ Recall []float64

+ RecallAvg float64

+ RecallWeightedAvg float64

+ func GetScores(confusionMatrix [][]int, sumI, sumJ []int, sumIJ, sumTrue int) (s Scores)

Mar 15, 2021

Changes in this version

type Config

+ Model ModelConfig

+ type Model interface

+ Learn func(texts dataset.DataSet, pipe *pipeline.Config) float32

+ Predict func(text string, pipe *pipeline.Config) (predictedClass string, proba float32)

+ Save func(directory string) error

+ type ModelConfig struct

+ Classifier string

+ Directory string

+ Load bool

+ Parameters map[string]interface{}

+ Save bool

+ WordVectorsConfig wordvectors.Config

Feb 22, 2021

Changes in this version

+ func Lower(text string) string

+ func Pipeline(text *string, pl *PipelineConfig) []string

+ func RemoveSymbols(text string) string

+ func Tokenize(text string) []string

+ type Classifier struct

+ Classes []bayesian.Class

+ Model *bayesian.Classifier

+ Pipeline PipelineConfig

+ func New(classifConfig *Config) *Classifier

+ func (c *Classifier) Predict(text string) (string, float64)

+ type Config struct

+ Classification []TrainingTexts

+ Pipeline PipelineConfig

+ func LoadConfig(path string) (*Config, error)

+ type PipelineConfig struct

+ Lower bool

+ RemoveSymbols bool

+ Threshold float64

+ type TrainingTexts struct

+ Command string

+ Texts []string