url/torchrln_url_ctgan.model/args.json

{
    "name": "torchrln",
    "objective": "classification",
    "batch_size": 1024,
    "epochs": 100,
    "early_stopping_rounds": 100,
    "learning_rate": 0.000844039801175779,
    "val_batch_size": 2048,
    "class_weight": "balanced",
    "weight_decay": 0,
    "force_device": "cpu",
    "is_text": false,
    "dg_method": null,
    "seed": 42,
    "dg_num_groups": 4,
    "dg_group_method": "random",
    "num_classes": 2,
    "n_layers": 2,
    "hidden_dim": 92,
    "norm": 1,
    "theta": -12,
    "custom_dataloader": "",
    "depth": 12,
    "heads": 2,
    "dropout": 0.5,
    "model_name": "torchrln",
    "dataset": "url",
    "num_splits": 5,
    "shuffle": true,
    "metrics": [
        "auc"
    ]
}