bert

Documentation

Index

• type QuestionAnswering
  • func LoadQuestionAnswering(modelPath string) (*QuestionAnswering, error)
  • func (qa *QuestionAnswering) ExtractAnswer(_ context.Context, question string, passage string, ...) (questionanswering.Response, error)

Types

type QuestionAnswering

type QuestionAnswering struct {
	// Model is the model used to answer questions.
	Model *bert.ModelForQuestionAnswering
	// Tokenizer is the tokenizer used to tokenize questions and passages.
	Tokenizer *wordpiecetokenizer.WordPieceTokenizer
}

QuestionAnswering is a QuestionAnswering model.

func LoadQuestionAnswering

func LoadQuestionAnswering(modelPath string) (*QuestionAnswering, error)

LoadQuestionAnswering returns a QuestionAnswering loading the model, the embeddings and the tokenizer from a directory.

func (*QuestionAnswering) ExtractAnswer

func (qa *QuestionAnswering) ExtractAnswer(_ context.Context, question string, passage string, opts *questionanswering.Options) (questionanswering.Response, error)

ExtractAnswer returns the answers for the given question and passage. The options may assume default values if those are not set.