refactor(steps): move feature generation/selection

jitingxu1 · May 13, 2024 · 3747650 · 3747650
1 parent 958f733
commit 3747650
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 2 deletions.
diff --git a/ibis_ml/steps/__init__.py b/ibis_ml/steps/__init__.py
@@ -6,9 +6,9 @@
     OneHotEncode,
     TargetEncode,
 )
-from ibis_ml.steps._feature_engineering import CreatePolynomialFeatures
-from ibis_ml.steps._feature_selection import DropZeroVariance
+from ibis_ml.steps._generate_features import CreatePolynomialFeatures
 from ibis_ml.steps._impute import FillNA, ImputeMean, ImputeMedian, ImputeMode
+from ibis_ml.steps._select_features import DropZeroVariance
 from ibis_ml.steps._standardize import ScaleMinMax, ScaleStandard
 from ibis_ml.steps._temporal import ExpandDate, ExpandDateTime, ExpandTime
 

diff --git a/ibis_ml/steps/_feature_engineering.py → ibis_ml/steps/_generate_features.py b/ibis_ml/steps/_feature_engineering.py → ibis_ml/steps/_generate_features.py
diff --git a/ibis_ml/steps/_feature_selection.py → ibis_ml/steps/_select_features.py b/ibis_ml/steps/_feature_selection.py → ibis_ml/steps/_select_features.py
diff --git a/tests/test_feature_engineering.py → tests/test_generate_features.py b/tests/test_feature_engineering.py → tests/test_generate_features.py
diff --git a/tests/test_feature_selection.py → tests/test_select_features.py b/tests/test_feature_selection.py → tests/test_select_features.py