18 references to KMeansUtils
Microsoft.ML.KMeansClustering (18)
KMeansPlusPlusTrainer.cs (18)
256KMeansUtils.VerifyModelConsistency(centroids);
263KMeansUtils.VerifyModelConsistency(centroids);
457public readonly KMeansUtils.RowIndexGetter RowIndexGetter;
550public KMeansUtils.RowIndexGetter RowIndexGetter { get { return _acceleratedRowMap.RowIndexGetter; } }
745KMeansUtils.FindBestCluster(in point, clusters, clustersL2s, clusterCount, needRealDistanceSquared,
843Heap<KMeansUtils.WeightedPoint>[] buffer = null;
844var rowStats = KMeansUtils.ParallelWeightedReservoirSample(host, numThreads, 1, cursorFactory,
874KMeansUtils.WeightFunc weightFn = (in VBuffer<float> point, int pointRowIndex) =>
888KMeansUtils.ParallelWeightedReservoirSample(host, numThreads, numSamplesPerRound, cursorFactory, weightFn,
915KMeansUtils.ParallelMapReduce<float[], float[]>(
925int debugBestCluster = KMeansUtils.FindBestCluster(in point, clusters, clustersL2s);
942KMeansUtils.ParallelMapReduce<float[], float[]>(
945(ref VBuffer<float> point, int discard, float[] weights, Random rand) => weights[KMeansUtils.FindBestCluster(in point, clusters, clustersL2s)]++,
998Heap<KMeansUtils.WeightedPoint>[] buffer = null;
1000var rowStats = KMeansUtils.ParallelWeightedReservoirSample(host, numThreads, k, cursorFactory,
1244public KMeansUtils.RowIndexGetter RowIndexGetter => _acceleratedRowMap.RowIndexGetter;
1445KMeansUtils.VerifyModelConsistency(centroids);
1513KMeansUtils.FindBestCluster(in cursor.Features, centroids, centroidL2s, k, false, out minDistance, out cluster, out secMinDistance, out secCluster);