diff --git a/calibration.py b/calibration.py index 8e0923f..1a291ea 100644 --- a/calibration.py +++ b/calibration.py @@ -1,7 +1,11 @@ import logging +import sys import numpy as np +logging.basicConfig(stream=sys.stdout) + + # This function is a sklearn.calibration.calibration_curve modification def calibration_curve(y_true, y_prob, *, n_bins=5, strategy="uniform"): y_true = np.array(y_true) @@ -23,6 +27,9 @@ def calibration_curve(y_true, y_prob, *, n_bins=5, strategy="uniform"): try: binids = np.digitize(y_prob, bins) - 1 + + raise ValueError + except Exception as e: np.set_printoptions(threshold=sys.maxsize) logging.info("=" * 40)