{ "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" ] }