Skip to content

Commit 838e63b

Browse files
Add coalescer
1 parent 294ac60 commit 838e63b

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

autoPyTorch/pipeline/tabular_classification.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@
1818
from autoPyTorch.pipeline.components.preprocessing.tabular_preprocessing.TabularColumnTransformer import (
1919
TabularColumnTransformer
2020
)
21+
from autoPyTorch.pipeline.components.preprocessing.tabular_preprocessing.coalescer import (
22+
CoalescerChoice
23+
)
2124
from autoPyTorch.pipeline.components.preprocessing.tabular_preprocessing.encoding.base_encoder_choice import (
2225
EncoderChoice
2326
)
@@ -246,6 +249,7 @@ def _get_pipeline_steps(self, dataset_properties: Optional[Dict[str, Any]],
246249

247250
steps.extend([
248251
("imputer", SimpleImputer(random_state=self.random_state)),
252+
("coalescer", CoalescerChoice(default_dataset_properties, random_state=self.random_state)),
249253
("encoder", EncoderChoice(default_dataset_properties, random_state=self.random_state)),
250254
("scaler", ScalerChoice(default_dataset_properties, random_state=self.random_state)),
251255
("feature_preprocessor", FeatureProprocessorChoice(default_dataset_properties,

autoPyTorch/pipeline/tabular_regression.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@
1616
from autoPyTorch.pipeline.components.preprocessing.tabular_preprocessing.TabularColumnTransformer import (
1717
TabularColumnTransformer
1818
)
19+
from autoPyTorch.pipeline.components.preprocessing.tabular_preprocessing.coalescer import (
20+
CoalescerChoice
21+
)
1922
from autoPyTorch.pipeline.components.preprocessing.tabular_preprocessing.encoding.base_encoder_choice import (
2023
EncoderChoice
2124
)
@@ -190,6 +193,7 @@ def _get_pipeline_steps(self, dataset_properties: Optional[Dict[str, Any]]) -> L
190193

191194
steps.extend([
192195
("imputer", SimpleImputer(random_state=self.random_state)),
196+
("coalescer", CoalescerChoice(default_dataset_properties, random_state=self.random_state)),
193197
("encoder", EncoderChoice(default_dataset_properties, random_state=self.random_state)),
194198
("scaler", ScalerChoice(default_dataset_properties, random_state=self.random_state)),
195199
("feature_preprocessor", FeatureProprocessorChoice(default_dataset_properties,

0 commit comments

Comments
 (0)