diff --git a/mastml/feature_selectors.py b/mastml/feature_selectors.py index 8e0493fa..4551a944 100644 --- 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') \ No newline at end of file + f.write(str(feature) + '\n')