word2vec

Documentation

Index

• Constants
• type Client
  • func New(opts ...Option) (*Client, error)
  • func (c *Client) ConsumedTokens() int
  • func (c *Client) Embedding(ctx context.Context, text string) ([]float32, error)
• type Option
  • func WithModel(path string) Option
  • func WithVectorSize(size int) Option

Constants

const Version = "word2vec/v0.1.0"

Types

type Client

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

func New

func New(opts ...Option) (*Client, error)

Creates word2vec embeddings client.

It requires configuration of the path to trained model

The client is configurable using

WithModel(path string)
WithVectorSize(size int)

func (*Client) ConsumedTokens

func (c *Client) ConsumedTokens() int

Number of tokens consumed within the session

func (*Client) Embedding

func (c *Client) Embedding(ctx context.Context, text string) ([]float32, error)

Calculates embedding vector

type Option

type Option func(*Client)

func WithModel

func WithModel(path string) Option

func WithVectorSize

func WithVectorSize(size int) Option