Merge branch 'master' of https://github.com/uw-cmg/MAST-ML

uw-cmg · Apr 17, 2024 · d0e2411 · d0e2411
2 parents 31e4084 + fbe73dc
commit d0e2411
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
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):

diff --git a/mastml/feature_selectors.py b/mastml/feature_selectors.py
@@ -34,7 +34,6 @@
 
 """
 
-import copy
 import os
 import warnings
 import shap
@@ -555,8 +554,8 @@ def fit(self, X, y):
 
             #### Print the removed features and the new smaller dataframe with features removed
             # Deep copy the original dataframe
-            removed_features_df = copy.deepcopy(X)
-            new_df = copy.deepcopy(X)
+            removed_features_df = copy(X)
+            new_df = copy(X)
 
             # Drop the features that can be removed
             all_features = list(new_df.columns)
@@ -910,4 +909,4 @@ def selected_features_correlation(X, savepath, features_x_path, features_y_path)
 
     with open(os.path.join(savepath, 'same_features.txt'), 'w') as f:
         for feature in same_features:
-            f.write(str(feature) + '\n')
+            f.write(str(feature) + '\n')