Expand description
KMeans implementation for Apache Arrow Arrays.
Support l2, cosine and dot distances, see DistanceType.
Cosine distance are calculated by normalizing the vectors to unit length,
and run l2 distance on the unit vectors.
Structs§
- KMeans
- KMeans implementation for Apache Arrow Arrays.
- KMeans
Algo Float - KMeans
Params - KMean Training Parameters
Enums§
- KMean
Init - KMean initialization method.
Traits§
Functions§
- compute_
partition - compute_
partitions - Compute partition ID of each vector in the KMeans.
- compute_
partitions_ arrow_ array - Compute partitions from Arrow FixedSizeListArray.
- kmeans_
find_ partitions - KMeans finds N nearest partitions.
- kmeans_
find_ partitions_ arrow_ array - kmeans_
find_ partitions_ binary