feature_transformer_type should be feature_transform_type

ATOMScience-org · Jan 22, 2025 · 1eda346 · 1eda346
1 parent 8feea36
commit 1eda346
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/atomsci/ddm/pipeline/featurization.py b/atomsci/ddm/pipeline/featurization.py
@@ -1606,7 +1606,7 @@ def create_feature_transformer(self, dataset, params):
         """
         if params.feature_transform_type == 'normalization':
             transformers_x = [trans.NormalizationTransformerMissingData(transform_X=True, dataset=dataset)]
-        elif params.feature_transformer_type == 'RobustScaler':
+        elif params.feature_transform_type == 'RobustScaler':
             transformers_x = [
                 trans.SklearnTransformerWrapper(transform_X=True, dataset=dataset,
                     sklearn_transformer=RobustScaler(
@@ -1616,7 +1616,7 @@ def create_feature_transformer(self, dataset, params):
                         unit_variance=params.robustscaler_unit_variance
                     ))
                 ]
-        elif params.feature_transformer_type == 'PowerTransformer':
+        elif params.feature_transform_type == 'PowerTransformer':
             transformers_x = [
                 trans.SklearnTransformerWrapper(transform_X=True, dataset=dataset,
                     sklearn_transformer=PowerTransformer(