predictor

Documentation

Index

• type Predictor
  • func NewPredictor(tokenizer TokenizerInterface, tokenRepository TokenRepositoryInterface, ...) *Predictor
  • func (p *Predictor) Predict(brainID int64, text string) (prediction entity.Prediction, err error)
• type ResultsRepositoryInterface
• type TokenRepositoryInterface
• type TokenizerInterface

Types

type Predictor

type Predictor struct {
	// contains filtered or unexported fields
}

Predictor is a structure which does tokenization and pass tokenized input with trained model to Classifier

func NewPredictor

func NewPredictor(
	tokenizer TokenizerInterface,
	tokenRepository TokenRepositoryInterface,
	resultsRepository ResultsRepositoryInterface,
) *Predictor

NewPredictor returns a Predictor instance

func (*Predictor) Predict

func (p *Predictor) Predict(brainID int64, text string) (prediction entity.Prediction, err error)

Predict returns prediction based on trained model provided by specified brain

type ResultsRepositoryInterface

type ResultsRepositoryInterface interface {
	GetTrainedModel(brainID int64) (classifier.TrainedModelInterface, error)
}

type TokenRepositoryInterface

type TokenRepositoryInterface interface {
	GetTokenIDs(brainID int64, tokens []string) ([]int64, error)
}

type TokenizerInterface

type TokenizerInterface interface {
	Tokenize(sentence string) []string
}