| { |
| "name": "tabtransformer", |
| "objective": "classification", |
| "batch_size": 1024, |
| "epochs": 100, |
| "early_stopping_rounds": 100, |
| "learning_rate": -4, |
| "val_batch_size": 2048, |
| "class_weight": "balanced", |
| "weight_decay": -3, |
| "force_device": "cpu", |
| "is_text": false, |
| "dg_method": null, |
| "seed": 42, |
| "dg_num_groups": 4, |
| "dg_group_method": "random", |
| "num_classes": 2, |
| "dim": 32, |
| "depth": 1, |
| "heads": 8, |
| "dropout": 0.2, |
| "custom_dataloader": "", |
| "model_name": "tabtransformer", |
| "dataset": "wids", |
| "num_splits": 5, |
| "shuffle": true, |
| "metrics": [ |
| "auc" |
| ] |
| } |
