ivf

type IVFIndex struct {
	NList     int            // Number of inverted lists
	NProbe    int            // Number of lists to search
	KMeans    *KMeans        // K-means clusterer for routing
	Lists     [][]int        // Inverted lists (vector indices)
	Vectors   []simd.Vec512  // All vectors
	Scales    []float32      // Vector scales
	IDs       []int          // External IDs
	ListLocks []sync.RWMutex // Per-list locks for concurrent access
	Trained   bool
}

type IVFIndexOptimized struct {
	NList     int              // Number of inverted lists
	NProbe    int              // Number of lists to search
	KMeans    *KMeansOptimized // Optimized K-means clusterer for routing
	Lists     [][]int          // Inverted lists (vector indices)
	Vectors   []simd.Vec512    // All vectors
	Scales    []float32        // Vector scales
	IDs       []int            // External IDs
	ListLocks []sync.RWMutex   // Per-list locks for concurrent access
	Trained   bool
	// contains filtered or unexported fields
}

type IndexStats struct {
	TotalVectors    int
	NumLists        int
	MinListSize     int
	MaxListSize     int
	AvgListSize     float64
	ListImbalance   float64 // Higher = more imbalanced
	MemoryUsage     int64   // Bytes
	TrainingVectors int
}

type KMeans struct {
	K         int           // Number of clusters
	MaxIters  int           // Maximum iterations
	Centroids []simd.Vec512 // Cluster centroids (quantized)
	Scales    []float32     // Centroid scales for dequantization
	Counts    []int         // Number of points per cluster
}

type KMeansOptimized struct {
	K         int           // Number of clusters
	MaxIters  int           // Maximum iterations
	Centroids []simd.Vec512 // Cluster centroids (quantized)
	Scales    []float32     // Centroid scales for dequantization
	Counts    []int         // Number of points per cluster
	// contains filtered or unexported fields
}

type SearchStats struct {
	ClustersSearched int
	CandidatesScored int
	ResultsReturned  int
}