Fix FutureWarning

INRIA · Jul 16, 2024 · 4dcd7fd · 4dcd7fd
1 parent 68c459b
commit 4dcd7fd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python_scripts/metrics_classification.py b/python_scripts/metrics_classification.py
@@ -347,7 +347,9 @@
 # of the positive class).
 
 # %%
-prevalence = target_test.value_counts()[1] / target_test.value_counts().sum()
+prevalence = (
+    target_test.value_counts()["donated"] / target_test.value_counts().sum()
+)
 print(f"Prevalence of the class 'donated': {prevalence:.2f}")
 
 # %% [markdown]