Hugging Face's logo

serval-uni-lu
/
tabularbench

Model card Files
xet
 Community
tabularbench
File size: 742 Bytes
b6f2ac4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
{
    "name": "tabnet",
    "objective": "classification",
    "batch_size": 1024,
    "epochs": 100,
    "early_stopping_rounds": 100,
    "learning_rate": -3,
    "val_batch_size": 1024,
    "class_weight": "balanced",
    "weight_decay": -3,
    "force_device": "cuda",
    "is_text": false,
    "num_classes": 2,
    "n_d": 11,
    "n_steps": 3,
    "gamma": 1.582741958275748,
    "cat_emb_dim": 3,
    "n_independent": 3,
    "n_shared": 4,
    "momentum": 0.2060205097281453,
    "mask_type": "entmax",
    "custom_dataloader": "",
    "depth": 12,
    "heads": 2,
    "dropout": 0.5,
    "model_name": "tabnet",
    "dataset": "malware",
    "num_splits": 5,
    "seed": 42,
    "shuffle": true,
    "metrics": [
        "auc"
    ]
}