Skip to content

Commit 680d37f

Browse files
authored
Use auto for tabnet (#2150)
1 parent 48731ae commit 680d37f

1 file changed

Lines changed: 1 addition & 8 deletions

File tree

ludwig/automl/defaults/combiner/tabnet_config.yaml

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,26 +2,19 @@ combiner:
22
type: tabnet
33

44
trainer:
5-
# validation_metric: accuracy
6-
batch_size: auto #2048
7-
learning_rate: auto #0.02
5+
batch_size: auto
86
learning_rate_scaling: sqrt
97
decay: true
108
decay_steps: 20000
119
decay_rate: 0.8
12-
# early_stop: 300
1310
optimizer:
1411
type: adam
1512

1613
hyperopt:
17-
# goal: maximize
1814
parameters:
1915
trainer.learning_rate:
2016
space: choice
2117
categories: [0.005, 0.01, 0.02, 0.025]
22-
trainer.batch_size:
23-
space: choice
24-
categories: [256, 512, 1024, 2048, 4096, 8192]
2518
trainer.decay_rate:
2619
space: choice
2720
categories: [0.8, 0.9, 0.95]

0 commit comments

Comments
 (0)