Fixed calibration_curve's normalize deprecated in sklearn

Broundal · May 6, 2024 · 7b866cd · 7b866cd
1 parent 78bc024
commit 7b866cd
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/pytolemaic/analysis_logic/model_analysis/scoring/scoring_report.py b/pytolemaic/analysis_logic/model_analysis/scoring/scoring_report.py
@@ -159,7 +159,6 @@ def __init__(self, y_true, y_proba, labels=None, sample_weight=None,
                 sklearn.calibration.calibration_curve(
                     y_true=y_true == class_index,
                     y_prob=y_proba[:, class_index],
-                    normalize=False,
                     n_bins=n_bins,
                     strategy='uniform')
             self._calibration_curve[label] = dict(

diff --git a/pytolemaic/prediction_uncertainty/uncertainty_model.py b/pytolemaic/prediction_uncertainty/uncertainty_model.py
@@ -480,7 +480,7 @@ def fit_uncertainty_model(self, dmd_test, n_jobs=multiprocessing.cpu_count() - 1
             self._fraction_of_positives, self._mean_uncertainty = sklearn.calibration.calibration_curve(
                 y_true=y_pred == y_true,
                 y_prob=uncertainty,
-                normalize=True,
+                # normalize=True - deprecated. not sure on implications
                 n_bins=self._n_bins,
                 strategy='uniform')
 

diff --git a/pytolemaic/version.py b/pytolemaic/version.py
@@ -1 +1 @@
-version = '0.15.4'
+version = '0.16.0'