Merge branch 'main' into upgrade-dependencies

probabl-ai · Feb 24, 2025 · c3b3472 · c3b3472
2 parents 7fa0880 + 01dc294
commit c3b3472
Show file tree

Hide file tree

Showing 19 changed files with 2,074 additions and 41 deletions.
diff --git a/examples/getting_started/plot_skore_getting_started.py b/examples/getting_started/plot_skore_getting_started.py
@@ -162,25 +162,34 @@
 # %%
 from skore import ComparisonReport
 
-comparator = ComparisonReport(reports=[log_reg_report, rf_report])
+comparison_report = ComparisonReport(reports=[log_reg_report, rf_report])
 
 # %%
 # As for the :class:`~skore.EstimatorReport` and the
 # :class:`~skore.CrossValidationReport`, we have a helper:
 
 # %%
-comparator.help()
+comparison_report.help()
 
 # %%
 # Let us display the result of our benchmark:
 
 # %%
-benchmark_metrics = comparator.metrics.report_metrics()
+benchmark_metrics = comparison_report.metrics.report_metrics()
 benchmark_metrics
 
 # %%
 # We have the result of our benchmark.
 
+# %%
+# We display the ROC curve for the two estimator reports we want to compare, by
+# superimposing them on the same figure:
+
+# %%
+comparison_report.metrics.roc().plot()
+plt.tight_layout()
+
+
 # %%
 # Train-test split with skore
 # ^^^^^^^^^^^^^^^^^^^^^^^^^^^

diff --git a/skore/src/skore/sklearn/__init__.py b/skore/src/skore/sklearn/__init__.py
@@ -11,11 +11,11 @@
 from skore.sklearn.train_test_split.train_test_split import train_test_split
 
 __all__ = [
-    "train_test_split",
+    "ComparisonReport",
     "CrossValidationReport",
     "EstimatorReport",
-    "ComparisonReport",
-    "RocCurveDisplay",
     "PrecisionRecallCurveDisplay",
     "PredictionErrorDisplay",
+    "RocCurveDisplay",
+    "train_test_split",
 ]