From 4c8b142db3a1a93179813813b35f3985cb83fc66 Mon Sep 17 00:00:00 2001 From: yagudin Date: Thu, 3 Jun 2021 01:04:52 +0300 Subject: [PATCH] Improve exception handling --- calibration.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/calibration.py b/calibration.py index 1013238..1a546cf 100644 --- a/calibration.py +++ b/calibration.py @@ -27,11 +27,12 @@ def calibration_curve(y_true, y_prob, *, n_bins=5, strategy="uniform"): try: binids = np.digitize(y_prob, bins) - 1 + except Exception as e: np.set_printoptions(threshold=sys.maxsize) logging.info("=" * 40) logging.info(f"n_bins={n_bins}, strategy={strategy}") - logging.info(f"y_true={repr(y_prob)}") + logging.info(f"y_prob={repr(y_prob)}") logging.info(f"bins={repr(bins)}") logging.info("=" * 40)