soci-agent-nn / config.json
RayMelius's picture
Train: acc=63.4%, 30 epochs
1d82fc7 verified
{
"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
}