{
  "model_type": "CompactGenderClassifier",
  "architectures": [
    "CompactGenderClassifier"
  ],
  "vocab_size": 4000,
  "emb_dim": 64,
  "hidden_dim": 128,
  "num_layers": 2,
  "num_heads": 4,
  "dim_feedforward": 256,
  "max_length": 20,
  "dropout": 0.1,
  "id2label": {
    "0": "MALE",
    "1": "FEMALE"
  },
  "label2id": {
    "MALE": 0,
    "FEMALE": 1
  },
  "metrics": {
    "best_val_accuracy": 0.8449828995420555,
    "best_val_f1": 0.8449560904215879
  }
}