#660 Enable Power Transformations Update (#1086)

* Power Transformer

* Correct typo

* ADD_630

* PEP8 compliance

* Fix target type

Co-authored-by: MaxGreil <[email protected]>

Author: franchuterivera

autosklearn/pipeline/components/data_preprocessing/rescaling/power_transformer.py

@@ -0,0 +1,32 @@
+from autosklearn.pipeline.constants import DENSE, UNSIGNED_DATA, INPUT
+from autosklearn.pipeline.components.data_preprocessing.rescaling.abstract_rescaling \
+    import Rescaling
+from autosklearn.pipeline.components.base import AutoSklearnPreprocessingAlgorithm
+
+
+class PowerTransformerComponent(Rescaling, AutoSklearnPreprocessingAlgorithm):
+    def __init__(self, random_state):
+        from sklearn.preprocessing import PowerTransformer
+        self.preprocessor = PowerTransformer(copy=False)
+
+    @staticmethod
+    def get_properties(dataset_properties=None):
+        return {'shortname': 'PowerTransformer',
+                'name': 'PowerTransformer',
+                'handles_missing_values': False,
+                'handles_nominal_values': False,
+                'handles_numerical_features': True,
+                'prefers_data_scaled': False,
+                'prefers_data_normalized': False,
+                'handles_regression': True,
+                'handles_classification': True,
+                'handles_multiclass': True,
+                'handles_multilabel': True,
+                'handles_multioutput': True,
+                'is_deterministic': True,
+                # TODO find out of this is right!
+                'handles_sparse': False,
+                'handles_dense': True,
+                'input': (DENSE, UNSIGNED_DATA),
+                'output': (INPUT,),
+                'preferred_dtype': None}

test/test_pipeline/test_classification.py

@@ -421,7 +421,7 @@ def test_get_hyperparameter_search_space(self):
         forbiddens = cs.get_forbiddens()
 
         self.assertEqual(len(cs.get_hyperparameter(
-            'data_preprocessing:numerical_transformer:rescaling:__choice__').choices), 6)
+            'data_preprocessing:numerical_transformer:rescaling:__choice__').choices), 7)
         self.assertEqual(len(cs.get_hyperparameter(
             'classifier:__choice__').choices), 16)
         self.assertEqual(len(cs.get_hyperparameter(