Change default threshold

rapidsai · Feb 26, 2025 · 36334a5 · 36334a5
1 parent 11054f7
commit 36334a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cpp/include/cuvs/preprocessing/quantize/binary.h b/cpp/include/cuvs/preprocessing/quantize/binary.h
@@ -37,7 +37,7 @@ struct cuvsBinaryQuantizerParams {
   /*
    * specifies the threshold to set a bit in cuvsBinaryQuantizerTransform
    */
-  cuvsBinaryQuantizerThreshold threshold = ZERO;
+  cuvsBinaryQuantizerThreshold threshold = MEAN;
 
   /*
    * specifies the sampling ratio

diff --git a/cpp/include/cuvs/preprocessing/quantize/binary.hpp b/cpp/include/cuvs/preprocessing/quantize/binary.hpp
@@ -40,7 +40,7 @@ enum class set_bit_threshold { zero, mean, sampling_median };
  * @brief quantizer parameters.
  */
 struct params {
-  set_bit_threshold threshold = set_bit_threshold::zero;
+  set_bit_threshold threshold = set_bit_threshold::mean;
 
   float sampling_ratio = 0.1;
 };