Skip to content

Commit e72c101

Browse files
Add coalescer
1 parent 018eeec commit e72c101

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 import (
2225
EncoderChoice
2326
)
@@ -254,6 +257,7 @@ def _get_pipeline_steps(self, dataset_properties: Optional[Dict[str, Any]],
254257

255258
steps.extend([
256259
("imputer", SimpleImputer(random_state=self.random_state)),
260+
("coalescer", CoalescerChoice(default_dataset_properties, random_state=self.random_state)),
257261
("encoder", EncoderChoice(default_dataset_properties, random_state=self.random_state)),
258262
("scaler", ScalerChoice(default_dataset_properties, random_state=self.random_state)),
259263
("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 import (
2023
EncoderChoice
2124
)
@@ -200,6 +203,7 @@ def _get_pipeline_steps(self, dataset_properties: Optional[Dict[str, Any]]) -> L
200203

201204
steps.extend([
202205
("imputer", SimpleImputer(random_state=self.random_state)),
206+
("coalescer", CoalescerChoice(default_dataset_properties, random_state=self.random_state)),
203207
("encoder", EncoderChoice(default_dataset_properties, random_state=self.random_state)),
204208
("scaler", ScalerChoice(default_dataset_properties, random_state=self.random_state)),
205209
("feature_preprocessor", FeatureProprocessorChoice(default_dataset_properties,

0 commit comments

Comments
 (0)