pq

package

v1.0.4 Latest Latest Go to latest Published: Mar 12, 2026 License: MIT Imports: 2 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Codebook struct {
	Centroids [][]float32
}

Codebook 码本

func NewCodebook(centroids [][]float32) *Codebook

NewCodebook 创建码本

func (cb *Codebook) Decode(index int) []float32

Decode 解码

func (cb *Codebook) Encode(vector []float32) int

Encode 编码

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

ProductQuantizer 乘积量化器

func NewProductQuantizer(dimension, m, k int) *ProductQuantizer

NewProductQuantizer 创建乘积量化器

func Quantize(vectors [][]float32, m int, k int) (*ProductQuantizer, error)

Quantize 量化向量

func (pq *ProductQuantizer) Decode(encoded []byte) []float32

Decode 解码向量

func (pq *ProductQuantizer) Encode(vector []float32) []byte

Encode 编码向量

func (pq *ProductQuantizer) Train(vectors [][]float32) error

Train 训练量化器