vectorizer

Documentation

Index

• func NewClassSettings(cfg moduletools.ClassConfig) *classSettings
• type ClassSettings
• type Client
• type Vectorizer
  • func New(client Client) *Vectorizer
  • func (v *Vectorizer) CombineVectors(vectors [][]float32) []float32
  • func (v *Vectorizer) MoveAwayFrom(source []float32, target []float32, weight float32) ([]float32, error)
  • func (v *Vectorizer) MoveTo(source []float32, target []float32, weight float32) ([]float32, error)
  • func (v *Vectorizer) Object(ctx context.Context, object *models.Object, objDiff *moduletools.ObjectDiff, ...) error
  • func (v *Vectorizer) Texts(ctx context.Context, inputs []string, settings ClassSettings) ([]float32, error)
  • func (v *Vectorizer) VectorizeImage(ctx context.Context, image string) ([]float32, error)

Functions

func NewClassSettings

func NewClassSettings(cfg moduletools.ClassConfig) *classSettings

Types

type ClassSettings

type ClassSettings interface {
	ImageField(property string) bool
	ImageFieldsWeights() ([]float32, error)
	TextField(property string) bool
	TextFieldsWeights() ([]float32, error)
}

type Client

type Client interface {
	Vectorize(ctx context.Context,
		texts, images []string) (*ent.VectorizationResult, error)
}

type Vectorizer

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

func New

func New(client Client) *Vectorizer

func (*Vectorizer) CombineVectors

func (v *Vectorizer) CombineVectors(vectors [][]float32) []float32

CombineVectors combines all of the vector into sum of their parts

func (*Vectorizer) MoveAwayFrom

func (v *Vectorizer) MoveAwayFrom(source []float32, target []float32, weight float32,
) ([]float32, error)

MoveAwayFrom moves one vector away from another

func (*Vectorizer) MoveTo

func (v *Vectorizer) MoveTo(source []float32, target []float32, weight float32,
) ([]float32, error)

MoveTo moves one vector toward another

func (*Vectorizer) Object

func (v *Vectorizer) Object(ctx context.Context, object *models.Object,
	objDiff *moduletools.ObjectDiff, settings ClassSettings,
) error

func (*Vectorizer) Texts

func (v *Vectorizer) Texts(ctx context.Context, inputs []string,
	settings ClassSettings,
) ([]float32, error)

func (*Vectorizer) VectorizeImage

func (v *Vectorizer) VectorizeImage(ctx context.Context, image string) ([]float32, error)