Update kdtree_knn_classification_train_dense_default_impl.i

oneapi-src · Sep 6, 2024 · e6528a6 · e6528a6
1 parent c99f49d
commit e6528a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...l/src/algorithms/k_nearest_neighbors/kdtree_knn_classification_train_dense_default_impl.i b/...l/src/algorithms/k_nearest_neighbors/kdtree_knn_classification_train_dense_default_impl.i
@@ -492,7 +492,7 @@ algorithmFpType KNNClassificationTrainBatchKernel<algorithmFpType, training::def
     algorithmFpType * subSamples, size_t subSampleCapacity, Status & status)
 {
     algorithmFpType samples[__KDTREE_MEDIAN_RANDOM_SAMPLE_COUNT + 1];
-    const size_t sampleCount = sizeof(samples) / sizeof(samples[0]);
+    const size_t sampleCount = __KDTREE_MEDIAN_RANDOM_SAMPLE_COUNT + 1;
 
     if (end - start <= sampleCount)
     {