Update __init__.py

vertica · Nov 1, 2023 · 8a3de82 · 8a3de82
1 parent 5fd2097
commit 8a3de82
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/verticapy/machine_learning/metrics/__init__.py b/verticapy/machine_learning/metrics/__init__.py
@@ -113,6 +113,7 @@
     "fm": fowlkes_mallows_index,
     "prevalence_threshold": prevalence_threshold,
     "pm": prevalence_threshold,
+    "confusion_matrix": confusion_matrix,
 }
 
 FUNCTIONS_REGRESSION_DICTIONNARY = {