{
  "architecture": "SociAgentTransformer",
  "d_model": 128,
  "nhead": 8,
  "num_layers": 4,
  "d_ff": 256,
  "num_experts": 4,
  "feature_dim": 47,
  "num_actions": 9,
  "num_locations": 40,
  "action_types": [
    "move",
    "work",
    "eat",
    "sleep",
    "talk",
    "exercise",
    "shop",
    "relax",
    "wander"
  ],
  "locations": [
    "house_elena",
    "house_marcus",
    "house_helen",
    "house_diana",
    "house_kai",
    "house_priya",
    "house_james",
    "house_rosa",
    "house_yuki",
    "house_frank",
    "apartment_block_1",
    "apartment_block_2",
    "apartment_block_3",
    "apt_northeast",
    "apt_northwest",
    "apt_southeast",
    "apt_southwest",
    "cafe",
    "grocery",
    "bar",
    "restaurant",
    "bakery",
    "cinema",
    "diner",
    "pharmacy",
    "office",
    "office_tower",
    "factory",
    "school",
    "hospital",
    "park",
    "gym",
    "library",
    "church",
    "town_square",
    "sports_field",
    "street_north",
    "street_south",
    "street_east",
    "street_west"
  ],
  "action_durations": {
    "move": 1,
    "work": 4,
    "eat": 2,
    "sleep": 8,
    "talk": 2,
    "exercise": 3,
    "shop": 2,
    "relax": 2,
    "wander": 1
  },
  "need_names": [
    "hunger",
    "energy",
    "social",
    "purpose",
    "comfort",
    "fun"
  ],
  "personality_names": [
    "openness",
    "conscientiousness",
    "extraversion",
    "agreeableness",
    "neuroticism"
  ],
  "best_val_action_acc": 0.6343636363636364,
  "training_samples": 99000,
  "epochs": 30
}