config.json

{
  "architectures": [
    "TabNetForClassification"
  ],
  "cat_dims": [],
  "cat_emb_dim": 1,
  "cat_idxs": [],
  "d_as": 64,
  "d_m": 64,
  "d_na": 64,
  "epsilon": 1e-15,
  "gamma": 1.3,
  "id2label": {
    "0": "Benign",
    "1": "DDoS",
    "2": "PortScan",
    "3": "Botnet",
    "4": "Infiltration"
  },
  "label2id": {
    "Benign": 0,
    "Botnet": 3,
    "DDoS": 1,
    "Infiltration": 4,
    "PortScan": 2
  },
  "lambda_sparse": 0.001,
  "model_type": "tabnet",
  "n_d": 64,
  "n_independent": 2,
  "n_shared": 2,
  "n_steps": 5,
  "num_features": 78,
  "torch_dtype": "float32",
  "transformers_version": "4.36.0"
}