File size: 656 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 | {
"name": "stg",
"objective": "classification",
"batch_size": 1024,
"epochs": 100,
"early_stopping_rounds": 100,
"learning_rate": 0.09855355322884699,
"val_batch_size": 1024,
"class_weight": "balanced",
"weight_decay": -3,
"force_device": "cuda",
"is_text": false,
"lam": 5.8725651364978955,
"hidden_dims": [
500,
500,
10
],
"num_classes": 2,
"custom_dataloader": "",
"depth": 12,
"heads": 2,
"dropout": 0.5,
"model_name": "stg",
"dataset": "malware",
"num_splits": 5,
"seed": 42,
"shuffle": true,
"metrics": [
"auc"
]
} |