Hugging Face's logo

RayMelius
/
soci-agent-nn

ONNX
Model card Files
xet
 Community
RayMelius commited on
Commit
1d82fc7
·
verified ·
1 Parent(s): b6f5914

Train: acc=63.4%, 30 epochs

Browse files
Files changed (1) hide show
  1. config.json +92 -92
config.json CHANGED
@@ -1,93 +1,93 @@
1
- {
2
- "architecture": "SociAgentTransformer",
3
- "d_model": 128,
4
- "nhead": 8,
5
- "num_layers": 4,
6
- "d_ff": 256,
7
- "num_experts": 4,
8
- "feature_dim": 47,
9
- "num_actions": 9,
10
- "num_locations": 40,
11
- "action_types": [
12
- "move",
13
- "work",
14
- "eat",
15
- "sleep",
16
- "talk",
17
- "exercise",
18
- "shop",
19
- "relax",
20
- "wander"
21
- ],
22
- "locations": [
23
- "house_elena",
24
- "house_marcus",
25
- "house_helen",
26
- "house_diana",
27
- "house_kai",
28
- "house_priya",
29
- "house_james",
30
- "house_rosa",
31
- "house_yuki",
32
- "house_frank",
33
- "apartment_block_1",
34
- "apartment_block_2",
35
- "apartment_block_3",
36
- "apt_northeast",
37
- "apt_northwest",
38
- "apt_southeast",
39
- "apt_southwest",
40
- "cafe",
41
- "grocery",
42
- "bar",
43
- "restaurant",
44
- "bakery",
45
- "cinema",
46
- "diner",
47
- "pharmacy",
48
- "office",
49
- "office_tower",
50
- "factory",
51
- "school",
52
- "hospital",
53
- "park",
54
- "gym",
55
- "library",
56
- "church",
57
- "town_square",
58
- "sports_field",
59
- "street_north",
60
- "street_south",
61
- "street_east",
62
- "street_west"
63
- ],
64
- "action_durations": {
65
- "move": 1,
66
- "work": 4,
67
- "eat": 2,
68
- "sleep": 8,
69
- "talk": 2,
70
- "exercise": 3,
71
- "shop": 2,
72
- "relax": 2,
73
- "wander": 1
74
- },
75
- "need_names": [
76
- "hunger",
77
- "energy",
78
- "social",
79
- "purpose",
80
- "comfort",
81
- "fun"
82
- ],
83
- "personality_names": [
84
- "openness",
85
- "conscientiousness",
86
- "extraversion",
87
- "agreeableness",
88
- "neuroticism"
89
- ],
90
- "best_val_action_acc": 0.6084,
91
- "training_samples": 100000,
92
- "epochs": 30
93
  }
 
1
+ {
2
+ "architecture": "SociAgentTransformer",
3
+ "d_model": 128,
4
+ "nhead": 8,
5
+ "num_layers": 4,
6
+ "d_ff": 256,
7
+ "num_experts": 4,
8
+ "feature_dim": 47,
9
+ "num_actions": 9,
10
+ "num_locations": 40,
11
+ "action_types": [
12
+ "move",
13
+ "work",
14
+ "eat",
15
+ "sleep",
16
+ "talk",
17
+ "exercise",
18
+ "shop",
19
+ "relax",
20
+ "wander"
21
+ ],
22
+ "locations": [
23
+ "house_elena",
24
+ "house_marcus",
25
+ "house_helen",
26
+ "house_diana",
27
+ "house_kai",
28
+ "house_priya",
29
+ "house_james",
30
+ "house_rosa",
31
+ "house_yuki",
32
+ "house_frank",
33
+ "apartment_block_1",
34
+ "apartment_block_2",
35
+ "apartment_block_3",
36
+ "apt_northeast",
37
+ "apt_northwest",
38
+ "apt_southeast",
39
+ "apt_southwest",
40
+ "cafe",
41
+ "grocery",
42
+ "bar",
43
+ "restaurant",
44
+ "bakery",
45
+ "cinema",
46
+ "diner",
47
+ "pharmacy",
48
+ "office",
49
+ "office_tower",
50
+ "factory",
51
+ "school",
52
+ "hospital",
53
+ "park",
54
+ "gym",
55
+ "library",
56
+ "church",
57
+ "town_square",
58
+ "sports_field",
59
+ "street_north",
60
+ "street_south",
61
+ "street_east",
62
+ "street_west"
63
+ ],
64
+ "action_durations": {
65
+ "move": 1,
66
+ "work": 4,
67
+ "eat": 2,
68
+ "sleep": 8,
69
+ "talk": 2,
70
+ "exercise": 3,
71
+ "shop": 2,
72
+ "relax": 2,
73
+ "wander": 1
74
+ },
75
+ "need_names": [
76
+ "hunger",
77
+ "energy",
78
+ "social",
79
+ "purpose",
80
+ "comfort",
81
+ "fun"
82
+ ],
83
+ "personality_names": [
84
+ "openness",
85
+ "conscientiousness",
86
+ "extraversion",
87
+ "agreeableness",
88
+ "neuroticism"
89
+ ],
90
+ "best_val_action_acc": 0.6343636363636364,
91
+ "training_samples": 99000,
92
+ "epochs": 30
93
  }