Fix bug of CategoricalFocalLoss #20

Author: jackguagua

deeptables/models/layers.py

@@ -986,7 +986,7 @@ def __init__(self, gamma=2., alpha=.25, reduction=losses.Reduction.AUTO, name='f
             gamma {float} -- (default: {2.0})
             alpha {float} -- (default: {4.0})
         """
-        super(BinaryFocalLoss, self).__init__(reduction=reduction, name=name)
+        super(CategoricalFocalLoss, self).__init__(reduction=reduction, name=name)
         self.gamma = float(gamma)
         self.alpha = float(alpha)
 

deeptables/preprocessing/transformer.py

@@ -7,6 +7,10 @@
 from sklearn.utils import column_or_1d
 from ..utils import dt_logging, consts
 
+from sklearn.pipeline import Pipeline
+from sklearn.base import BaseEstimator,TransformerMixin
+from sklearn.pipeline import FeatureUnion
+
 logger = dt_logging.get_logger()