seq2seq

Documentation

Index

• type BartForConditionalGeneration
  • func LoadModel(modelPath string) (*BartForConditionalGeneration, error)
  • func (t *BartForConditionalGeneration) Generate(text string) (string, error)

Types

type BartForConditionalGeneration

type BartForConditionalGeneration struct {
	*conditionalgeneration.Model
	Tokenizer *sentencepiece.Tokenizer
}

BartForConditionalGeneration contains the Model and the Tokenizer used for conditional generation tasks. For example, Machine Translation and Summarization.

func LoadModel

func LoadModel(modelPath string) (*BartForConditionalGeneration, error)

LoadModel loads a BartForConditionalGeneration from file.

func (*BartForConditionalGeneration) Generate

func (t *BartForConditionalGeneration) Generate(text string) (string, error)

Generate generates new texts starting from the input.