gobert

Documentation

Index

• type Bert
  • func New(model string) (*Bert, error)
  • func (l *Bert) Embeddings(text string, opts ...PredictOption) ([]float32, error)
  • func (l *Bert) Free()
  • func (l *Bert) TokenEmbeddings(tokens []int, opts ...PredictOption) ([]float32, error)
• type PredictOption
  • func SetEmbeddingSize(es int) PredictOption
  • func SetThreads(threads int) PredictOption
• type PredictOptions
  • func NewPredictOptions(opts ...PredictOption) PredictOptions

Types

type Bert

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

func New

func New(model string) (*Bert, error)

func (*Bert) Embeddings

func (l *Bert) Embeddings(text string, opts ...PredictOption) ([]float32, error)

func (*Bert) Free

func (l *Bert) Free()

func (*Bert) TokenEmbeddings

func (l *Bert) TokenEmbeddings(tokens []int, opts ...PredictOption) ([]float32, error)

type PredictOption

type PredictOption func(p *PredictOptions)

func SetEmbeddingSize

func SetEmbeddingSize(es int) PredictOption

SetThreads sets the number of threads to use for text generation.

func SetThreads

func SetThreads(threads int) PredictOption

SetThreads sets the number of threads to use for text generation.

type PredictOptions

type PredictOptions struct {
	Threads       int
	EmbeddingSize int
}

var DefaultOptions PredictOptions = PredictOptions{
	Threads:       4,
	EmbeddingSize: 99999,
}

func NewPredictOptions

func NewPredictOptions(opts ...PredictOption) PredictOptions

Create a new PredictOptions object with the given options.