rename: grn feature selection, transform func, k -> top_k.

4paradigm · Mar 25, 2022 · 0a12f8b · 0a12f8b
1 parent b5359c3
commit 0a12f8b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autox/autox_competition/feature_selection/grn_feature_selection.py b/autox/autox_competition/feature_selection/grn_feature_selection.py
@@ -343,8 +343,8 @@ def fit(self, df, y,column_definition):
         self.feature2weight['weight'] = self.weights
         print(self.feature2weight)
 
-    def transform(self, df, k=10):
-        ind = list(np.argpartition(self.weights, -k)[-k:])
+    def transform(self, df, top_k=10):
+        ind = list(np.argpartition(self.weights, -top_k)[-top_k:])
         names = list(np.array(self.new_columns)[ind])
         self.selected_df = df.loc[:, names]
         return self.selected_df