From 3747650ca3082554324ea7d43b0974bbe9b95065 Mon Sep 17 00:00:00 2001 From: Deepyaman Datta Date: Fri, 10 May 2024 22:23:30 -0600 Subject: [PATCH] refactor(steps): move feature generation/selection --- ibis_ml/steps/__init__.py | 4 ++-- .../steps/{_feature_engineering.py => _generate_features.py} | 0 ibis_ml/steps/{_feature_selection.py => _select_features.py} | 0 ...{test_feature_engineering.py => test_generate_features.py} | 0 tests/{test_feature_selection.py => test_select_features.py} | 0 5 files changed, 2 insertions(+), 2 deletions(-) rename ibis_ml/steps/{_feature_engineering.py => _generate_features.py} (100%) rename ibis_ml/steps/{_feature_selection.py => _select_features.py} (100%) rename tests/{test_feature_engineering.py => test_generate_features.py} (100%) rename tests/{test_feature_selection.py => test_select_features.py} (100%) diff --git a/ibis_ml/steps/__init__.py b/ibis_ml/steps/__init__.py index c2e9383..9be548c 100644 --- 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 b/ibis_ml/steps/_generate_features.py similarity index 100% rename from ibis_ml/steps/_feature_engineering.py rename to ibis_ml/steps/_generate_features.py diff --git a/ibis_ml/steps/_feature_selection.py b/ibis_ml/steps/_select_features.py similarity index 100% rename from ibis_ml/steps/_feature_selection.py rename to ibis_ml/steps/_select_features.py diff --git a/tests/test_feature_engineering.py b/tests/test_generate_features.py similarity index 100% rename from tests/test_feature_engineering.py rename to tests/test_generate_features.py diff --git a/tests/test_feature_selection.py b/tests/test_select_features.py similarity index 100% rename from tests/test_feature_selection.py rename to tests/test_select_features.py