Skip to content

Commit cd08be6

Browse files
fix TypeError in feature pruning
1 parent 4d8a3e6 commit cd08be6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

interface/apis.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -280,7 +280,7 @@ def compute_feature_importance(train: pd.DataFrame, test: pd.DataFrame, theta1:
280280

281281
def compute_feature_correlation(train: pd.DataFrame, test: pd.DataFrame, theta2: float):
282282
nonlocal pruning_info
283-
features = train[set(numerical_features).intersection(set(pruning_info.keys()))]
283+
features = train[list(set(numerical_features).intersection(set(pruning_info.keys())))]
284284
if target in list(features.columns):
285285
features = train.drop(target, axis=1)
286286
features_to_discard = set()

0 commit comments

Comments
 (0)