Fix: Resolve #32.

georgian-io · Mar 6, 2023 · c518857 · c518857
1 parent 1465eed
commit c518857
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/multimodal_transformers/data/data_utils.py b/multimodal_transformers/data/data_utils.py
@@ -66,7 +66,7 @@ def change_name_func(x):
     def _one_hot(self):
         ohe = preprocessing.OneHotEncoder(sparse=False)
         ohe.fit(self.df[self.cat_feats].values)
-        self.feat_names = list(ohe.get_feature_names(self.cat_feats))
+        self.feat_names = list(ohe.get_feature_names_out(self.cat_feats))
         return ohe.transform(self.df[self.cat_feats].values)
 
     def fit_transform(self):