evalsuite-python 0.1.0b1__tar.gz → 0.1.0b2__tar.gz

{evalsuite_python-0.1.0b1 → evalsuite_python-0.1.0b2}/CHANGELOG.md

@@ -6,6 +6,12 @@ All notable changes to this project are documented here. The format follows
 
 ## [Unreleased]
 
+## [0.1.0b2]
+
+### Changed
+- `es.plot.calibration`: the legend now sits below the axes by default so it no longer covers the
+  curves; new `legend_loc` argument (`"below"` or any matplotlib location).
+
 ## [0.1.0b1]
 
 Feature-complete for 0.1.0.

{evalsuite_python-0.1.0b1 → evalsuite_python-0.1.0b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: evalsuite-python
-Version: 0.1.0b1
+Version: 0.1.0b2
 Summary: A unified Python framework for machine-learning, clinical, statistical, segmentation, object-detection, uncertainty, and model evaluation.
 Project-URL: Homepage, https://evalsuite-nine.vercel.app
 Project-URL: Documentation, https://evalsuite-nine.vercel.app/docs
@@ -62,7 +62,7 @@ Description-Content-Type: text/markdown
 EvalSuite brings classification and regression metrics (with clinical, statistical, segmentation and
 object-detection evaluation on the roadmap) into one consistent, validated, documented framework.
 
-> **Status: beta (0.1.0b1).** Feature-complete for 0.1.0; the stable release follows once this beta is verified.
+> **Status: beta (0.1.0b2).** Feature-complete for 0.1.0; the stable release follows once this beta is verified.
 
 ## Installation
 

{evalsuite_python-0.1.0b1 → evalsuite_python-0.1.0b2}/README.md

@@ -10,7 +10,7 @@
 EvalSuite brings classification and regression metrics (with clinical, statistical, segmentation and
 object-detection evaluation on the roadmap) into one consistent, validated, documented framework.
 
-> **Status: beta (0.1.0b1).** Feature-complete for 0.1.0; the stable release follows once this beta is verified.
+> **Status: beta (0.1.0b2).** Feature-complete for 0.1.0; the stable release follows once this beta is verified.
 
 ## Installation
 

{evalsuite_python-0.1.0b1 → evalsuite_python-0.1.0b2}/src/evalsuite/plot.py

@@ -190,9 +190,13 @@ def calibration(
     strategy: str = "uniform",
     pos_label: Any = None,
     sample_weight: Optional[ArrayLike] = None,
+    legend_loc: str = "below",
 ) -> Axes:
     """Reliability diagram: observed frequency against mean predicted probability per bin, with ECE and Brier
-    score in the legend; the diagonal is perfect calibration."""
+    score in the legend; the diagonal is perfect calibration.
+
+    ``legend_loc="below"`` (default) puts the legend under the axes so it never covers the curves; any
+    matplotlib location (e.g. ``"upper left"``) places it inside instead."""
     from .classification.metrics import brier_score, calibration_curve, expected_calibration_error
 
     ax = _axes(ax)
@@ -218,7 +222,10 @@ def calibration(
         ylim=(-0.01, 1.01),
     )
     ax.set_aspect("equal")
-    ax.legend(loc="upper left", frameon=False)
+    if legend_loc == "below":
+        ax.legend(loc="upper center", bbox_to_anchor=(0.5, -0.14), frameon=False, fontsize="small")
+    else:
+        ax.legend(loc=legend_loc, frameon=False)  # type: ignore[call-overload]
     return ax
 
 

{evalsuite_python-0.1.0b1 → evalsuite_python-0.1.0b2}/src/evalsuite/version.py

@@ -1,3 +1,3 @@
 """Package version (single source of truth, read by the build backend)."""
 
-__version__ = "0.1.0b1"
+__version__ = "0.1.0b2"

{evalsuite_python-0.1.0b1 → evalsuite_python-0.1.0b2}/tests/output/test_plot.py

@@ -121,3 +121,13 @@ def test_matplotlib_is_optional(monkeypatch) -> None:
     monkeypatch.setattr(builtins, "__import__", fake)
     with pytest.raises(es.OptionalDependencyError, match=r"evalsuite-python\[plot\]"):
         es.plot.roc([0, 1], [0.2, 0.8])
+
+
+def test_calibration_legend_below_by_default_and_configurable():
+    y = np.array([0, 1, 0, 1, 1, 0, 1, 0])
+    p = np.array([0.1, 0.8, 0.3, 0.7, 0.9, 0.2, 0.6, 0.4])
+    ax = es.plot.calibration(y, p)
+    anchor = ax.get_legend().get_bbox_to_anchor().transformed(ax.transAxes.inverted())
+    assert anchor.y0 < 0  # outside, under the axes
+    ax2 = es.plot.calibration(y, p, legend_loc="upper left")
+    assert ax2.get_legend()._loc == 2