Skip to content

Commit 7450562

Browse files
committed
feat: add AutoML classifiers and enable test mode
- Add AutoGluon, TPOT, FLAML, and AutoKeras classifiers to `config_hyperopt.yml` and `config_single_run.yml`. - Enable `test_mode` in `config_hyperopt.yml` to use minimal parameter spaces for faster testing.
1 parent 42e67df commit 7450562

2 files changed

Lines changed: 10 additions & 0 deletions

File tree

config_hyperopt.yml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ global_params:
1313
sub_sample_param_space_pct: 0.05
1414
force_second_cv: false # If True, forces a second cross-validation run even if cached results are available. Defaults to False.
1515
model_eval_time_limit: 3600
16+
test_mode: true # If True, uses minimal parameter spaces and reduced cross-validation for fast testing.
1617

1718
# Experiment settings for the hyperopt run
1819
experiment:
@@ -62,7 +63,12 @@ models:
6263
# Set the following to true if a GPU is available and configured
6364
kerasClassifier_class: false
6465
knn__gpu_wrapper_class: false
66+
# Additional
6567
TabPFNClassifierClass: false # requires hf token and agreement
68+
AutoGluonClassifierClass: false # disabled by default
69+
TPOTClassifierClass: false # disabled by default
70+
FLAMLClassifierClass: false # disabled by default
71+
AutoKerasClassifierClass: false # disabled by default
6672

6773

6874
# This section defines the parameter search space for Hyperopt.

config_single_run.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,10 @@ models:
4747
H2O_GAM_class: true
4848
knn__gpu_wrapper_class: false
4949
TabPFNClassifierClass: false # requires hf token and agreement
50+
AutoGluonClassifierClass: false
51+
TPOTClassifierClass: false
52+
FLAMLClassifierClass: false
53+
AutoKerasClassifierClass: false
5054

5155
# This section defines a single set of parameters for a standalone run.
5256
run_params:

0 commit comments

Comments
 (0)