File size: 603 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 | {
"name": "saint",
"objective": "classification",
"batch_size": 128,
"epochs": 100,
"early_stopping_rounds": 100,
"learning_rate": -3,
"val_batch_size": 128,
"class_weight": "balanced",
"weight_decay": -3,
"force_device": "cuda",
"is_text": false,
"num_classes": 2,
"p_m": 0.3,
"K": 3,
"alpha": 2,
"beta": 1,
"custom_dataloader": "",
"depth": 12,
"heads": 2,
"dropout": 0.5,
"model_name": "vime",
"dataset": "malware",
"num_splits": 5,
"seed": 42,
"shuffle": true,
"metrics": [
"auc"
]
} |