We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ea510d6 commit f9957a5Copy full SHA for f9957a5
tabarena/tabarena/models/prep_lgb/generate.py
@@ -66,13 +66,13 @@ def generate_configs_lightgbm(num_random_configs=200) -> list:
66
r_num += 1
67
if configs[i].pop('use_arithmetic_preprocessor') == True:
68
configs[i]['prep_params'].update({
69
- 'ArithmeticPreprocessor': {}
+ 'ArithmeticFeatureGenerator': {}
70
})
71
72
if configs[i].pop('use_cat_fe') == True:
73
74
- 'CatIntAdder': {},
75
- 'OOFTargetEncoder': {}
+ 'CategoricalInteractionFeatureGenerator': {},
+ 'OOFTargetEncodingFeatureGenerator': {}
76
77
78
return [convert_numpy_dtypes(config) for config in configs]
0 commit comments