open.source/gjo-calibration
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Testing error in production

Browse Source
This commit is contained in:
yagudin 2021-06-03 00:55:34 +03:00
parent d6c632a361
commit 24ca5a885c
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
calibration.py
View File

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