|
52 | 52 |
|
53 | 53 | from sklearn.model_selection import cross_validate |
54 | 54 | from sklearn.pipeline import make_pipeline |
55 | | -from sklearn.compose import ColumnTransformer |
| 55 | +from sklearn.compose import make_column_transformer |
56 | 56 | from sklearn.preprocessing import OrdinalEncoder |
57 | 57 | from sklearn.ensemble import HistGradientBoostingClassifier |
58 | 58 |
|
59 | 59 | categorical_preprocessor = OrdinalEncoder( |
60 | 60 | handle_unknown="use_encoded_value", unknown_value=-1 |
61 | 61 | ) |
62 | | -preprocessor = ColumnTransformer( |
63 | | - [("categorical", categorical_preprocessor, categorical_columns)], |
| 62 | +preprocessor = make_column_transformer( |
| 63 | + (categorical_preprocessor, categorical_columns), |
64 | 64 | remainder="passthrough", |
65 | 65 | ) |
66 | 66 |
|
| 67 | + |
67 | 68 | model = make_pipeline(preprocessor, HistGradientBoostingClassifier()) |
68 | 69 |
|
69 | 70 | start = time.time() |
|
90 | 91 |
|
91 | 92 | from sklearn.preprocessing import StandardScaler |
92 | 93 |
|
93 | | -preprocessor = ColumnTransformer( |
94 | | - [ |
95 | | - ("numerical", StandardScaler(), numerical_columns), |
96 | | - ( |
97 | | - "categorical", |
98 | | - OrdinalEncoder( |
99 | | - handle_unknown="use_encoded_value", unknown_value=-1 |
100 | | - ), |
101 | | - categorical_columns, |
102 | | - ), |
103 | | - ] |
| 94 | +preprocessor = make_column_transformer( |
| 95 | + (StandardScaler(), numerical_columns), |
| 96 | + ( |
| 97 | + OrdinalEncoder(handle_unknown="use_encoded_value", unknown_value=-1), |
| 98 | + categorical_columns, |
| 99 | + ), |
104 | 100 | ) |
105 | 101 |
|
106 | 102 | model = make_pipeline(preprocessor, HistGradientBoostingClassifier()) |
|
151 | 147 | categorical_preprocessor = OneHotEncoder( |
152 | 148 | handle_unknown="ignore", sparse_output=False |
153 | 149 | ) |
154 | | -preprocessor = ColumnTransformer( |
155 | | - [("one-hot-encoder", categorical_preprocessor, categorical_columns)], |
| 150 | +preprocessor = make_column_transformer( |
| 151 | + (categorical_preprocessor, categorical_columns), |
156 | 152 | remainder="passthrough", |
157 | 153 | ) |
158 | 154 |
|
|
0 commit comments