Update fslite/fs/multivariate.py

ypriverol · qodo-merge-pro[bot] · web-flow · commit d5cc9748b50b · 2024-09-25T13:27:08.000+01:00
Co-authored-by: codiumai-pr-agent-pro[bot] &lt;151058649+codiumai-pr-agent-pro[bot]@users.noreply.github.com&gt;
diff --git a/fslite/fs/multivariate.py b/fslite/fs/multivariate.py
@@ -166,7 +166,9 @@ def multivariate_correlation_selector(
         )
 
     # Select feature index to keep
-    selected_features = [i for i in features_indexes if i not in index_to_remove]
+    mask = np.ones(len(features_indexes), dtype=bool)
+    mask[list(index_to_remove)] = False
+    selected_features = np.array(features_indexes)[mask]
 
     return selected_features
 

Original file line number	Diff line number	Diff line change
`@@ -166,7 +166,9 @@ def multivariate_correlation_selector(`
`166`	`166`	`)`
`167`	`167`
`168`	`168`	`# Select feature index to keep`
`169`		`- selected_features = [i for i in features_indexes if i not in index_to_remove]`
	`169`	`+ mask = np.ones(len(features_indexes), dtype=bool)`
	`170`	`+ mask[list(index_to_remove)] = False`
	`171`	`+ selected_features = np.array(features_indexes)[mask]`
`170`	`172`
`171`	`173`	`return selected_features`
`172`	`174`