nba

Documentation

Overview

Package nba provides facilities for vectorizing NBA data.

Index

• func InsertPlayers(ctx context.Context, conn *pgx.Conn, ps []Player) error
• type Player
  • func NearestPlayer(ctx context.Context, c *api.Client, conn *pgx.Conn, question string) (Player, error)
  • func NewPlayer(fields, row []string) (Player, error)
  • func (p *Player) GenerateEmbeddings(ctx context.Context, c *api.Client) error
  • func (p Player) String() string

Functions

func InsertPlayers

func InsertPlayers(ctx context.Context, conn *pgx.Conn, ps []Player) error

InsertPlayers inserts players into a database.

Types

type Player

type Player struct {
	Embedding pgvector.Vector
	// contains filtered or unexported fields
}

A Player represents an NBA player.

func NearestPlayer

func NearestPlayer(ctx context.Context, c *api.Client, conn *pgx.Conn, question string) (Player, error)

NearestPlayer returns the player whose embedding is closest to the given question.

func NewPlayer

func NewPlayer(fields, row []string) (Player, error)

NewPlayer returns a new Player from the given fields and row.

func (*Player) GenerateEmbeddings

func (p *Player) GenerateEmbeddings(ctx context.Context, c *api.Client) error

GenerateEmbeddings generates player embeddings.

func (Player) String

func (p Player) String() string