Skip to content

Commit f9957a5

Browse files
committed
bug fix in search space
1 parent ea510d6 commit f9957a5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tabarena/tabarena/models/prep_lgb/generate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,13 +66,13 @@ def generate_configs_lightgbm(num_random_configs=200) -> list:
6666
r_num += 1
6767
if configs[i].pop('use_arithmetic_preprocessor') == True:
6868
configs[i]['prep_params'].update({
69-
'ArithmeticPreprocessor': {}
69+
'ArithmeticFeatureGenerator': {}
7070
})
7171

7272
if configs[i].pop('use_cat_fe') == True:
7373
configs[i]['prep_params'].update({
74-
'CatIntAdder': {},
75-
'OOFTargetEncoder': {}
74+
'CategoricalInteractionFeatureGenerator': {},
75+
'OOFTargetEncodingFeatureGenerator': {}
7676
})
7777

7878
return [convert_numpy_dtypes(config) for config in configs]

0 commit comments

Comments
 (0)