Update feature_generators.py

Update polynomial features call for latest sklearn
uw-cmg · Apr 17, 2024 · a85a61a · a85a61a
1 parent 736eb08
commit a85a61a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mastml/feature_generators.py b/mastml/feature_generators.py
@@ -1421,7 +1421,7 @@ def fit(self, X, y=None):
         return self
 
     def transform(self, X):
-        new_features = self.SPF.get_feature_names()
+        new_features = self.SPF.get_feature_names_out()
         return pd.DataFrame(self.SPF.transform(self.featurize_df), columns=new_features), self.y
 
 class OneHotGroupGenerator(BaseGenerator):