update consts

Lightning-AI · Nov 3, 2022 · 1e5f130 · 1e5f130
1 parent 48188d5
commit 1e5f130
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/torchmetrics/classification/precision_recall_curve.py b/src/torchmetrics/classification/precision_recall_curve.py
@@ -46,10 +46,10 @@
 from torchmetrics.utilities.data import dim_zero_cat
 
 DYNAMIC_THRESHOLDS_MODE_STR = "dynamic"
-_DYNAMIC_THRESHOLDS_NBINS = 3 * 10**3
-_DYNAMIC_THRESHOLDS_MIN_NSAMPLES_PERBIN = 4 * 10**3
+_DYNAMIC_THRESHOLDS_NBINS = 10**4
+_DYNAMIC_THRESHOLDS_MIN_NSAMPLES_PERBIN = 10**4
 _DYNAMIC_THRESHOLDS_MIN_NSAMPLES = _DYNAMIC_THRESHOLDS_MIN_NSAMPLES_PERBIN * _DYNAMIC_THRESHOLDS_NBINS
-_DYNAMIC_THRESHOLDS_MAX_NSAMPLES_QUANTILE_ESTIMATION = 12 * 10**6
+_DYNAMIC_THRESHOLDS_MAX_NSAMPLES_QUANTILE_ESTIMATION = 10**7
 
 
 def _validate_budget(budget: Optional[int]) -> int: