| { |
| "name": "tabnet", |
| "objective": "classification", |
| "batch_size": 1024, |
| "epochs": 100, |
| "early_stopping_rounds": 100, |
| "learning_rate": -3, |
| "val_batch_size": 2048, |
| "class_weight": "balanced", |
| "weight_decay": -3, |
| "force_device": "cuda", |
| "is_text": false, |
| "num_classes": 2, |
| "n_d": 8, |
| "n_steps": 3, |
| "gamma": 1.2374567964327479, |
| "cat_emb_dim": 2, |
| "n_independent": 1, |
| "n_shared": 2, |
| "momentum": 0.009115552726326607, |
| "mask_type": "entmax", |
| "custom_dataloader": "", |
| "depth": 12, |
| "heads": 2, |
| "dropout": 0.5, |
| "model_name": "tabnet", |
| "dataset": "ctu_13_neris", |
| "num_splits": 5, |
| "seed": 42, |
| "shuffle": true, |
| "metrics": [ |
| "auc" |
| ] |
| } |
