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)