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