diff --git a/verticapy/machine_learning/metrics/__init__.py b/verticapy/machine_learning/metrics/__init__.py index 19a5b4110..1500c03c9 100755 --- 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 = {