Fix

tonywork · Apr 22, 2021 · 55f71b5 · 55f71b5
1 parent 0a7f8e5
commit 55f71b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pycaret/internal/metrics.py b/pycaret/internal/metrics.py
@@ -5,6 +5,7 @@
 class BinaryMulticlassScoreFunc:
     def __init__(self, score_func):
         self.score_func = score_func
+        self.__name__ = score_func.__name__
 
     def __call__(self, y_true, y_pred, **kwargs):
         if "average" in kwargs: