From 1e5f130c22c09f63d0cb07e9513c407a27168445 Mon Sep 17 00:00:00 2001 From: jpcbertoldo <24547377+jpcbertoldo@users.noreply.github.com> Date: Thu, 3 Nov 2022 01:02:40 +0100 Subject: [PATCH] update consts --- src/torchmetrics/classification/precision_recall_curve.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/torchmetrics/classification/precision_recall_curve.py b/src/torchmetrics/classification/precision_recall_curve.py index 2aacff9f772..0093f44cd70 100644 --- 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: