Todd Nief commited on
Commit
91e4cef
·
1 Parent(s): 3a10604

update stable model to be five col model

Browse files
Files changed (2) hide show
  1. config.json +110 -42
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -5,6 +5,7 @@
5
  "MultiTaskModel"
6
  ],
7
  "attention_dropout": 0.1,
 
8
  "columns": [
9
  "Unique Id",
10
  "Food Product Category",
@@ -22,47 +23,111 @@
22
  "Food Product Group",
23
  "Primary Food Product Group"
24
  ],
25
- "decoders": {
26
- "Level of Processing": {
27
- "0": "Culinary Ingredient",
28
- "1": "Freshly Prepared",
29
- "2": "Moderately Processed",
30
- "3": "Non-Food",
31
- "4": "Ultra-Processed",
32
- "5": "Whole/Minimally Processed"
33
- },
34
- "Primary Food Product Category": {
35
- "0": "Beef",
36
- "1": "Beverages",
37
- "2": "Bread, Grains & Legumes",
38
- "3": "Butter",
39
- "4": "Cheese",
40
- "5": "Chicken",
41
- "6": "Condiments & Snacks",
42
- "7": "Eggs",
43
- "8": "Fish (Farm-Raised)",
44
- "9": "Fish (Wild)",
45
- "10": "Fruit",
46
- "11": "Grain Products",
47
- "12": "Legumes",
48
- "13": "Meals",
49
- "14": "Meat",
50
- "15": "Milk",
51
- "16": "Milk & Dairy",
52
- "17": "Non-Food",
53
- "18": "Pork",
54
- "19": "Produce",
55
- "20": "Rice",
56
- "21": "Roots & Tuber",
57
- "22": "Roots & Tubers",
58
- "23": "Seafood",
59
- "24": "Tree Nuts & Seeds",
60
- "25": "Turkey, Other Poultry",
61
- "26": "Vegetables",
62
- "27": "Whole/Minimally Processed",
63
- "28": "Yogurt"
64
- }
65
- },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
66
  "dim": 768,
67
  "dropout": 0.1,
68
  "hidden_dim": 3072,
@@ -72,8 +137,11 @@
72
  "n_heads": 12,
73
  "n_layers": 6,
74
  "num_categories_per_task": [
 
75
  6,
76
- 29
 
 
77
  ],
78
  "pad_token_id": 0,
79
  "qa_dropout": 0.1,
 
5
  "MultiTaskModel"
6
  ],
7
  "attention_dropout": 0.1,
8
+ "classification": "mlp",
9
  "columns": [
10
  "Unique Id",
11
  "Food Product Category",
 
23
  "Food Product Group",
24
  "Primary Food Product Group"
25
  ],
26
+ "decoders": [
27
+ [
28
+ "Food Product Category",
29
+ {
30
+ "0": "Beef",
31
+ "1": "Beverages",
32
+ "10": "Fruit",
33
+ "11": "Grain Products",
34
+ "12": "Legumes",
35
+ "13": "Meals",
36
+ "14": "Meat",
37
+ "15": "Milk",
38
+ "16": "Milk & Dairy",
39
+ "17": "Non-Food",
40
+ "18": "Pork",
41
+ "19": "Produce",
42
+ "2": "Bread, Grains & Legumes",
43
+ "20": "Rice",
44
+ "21": "Roots & Tubers",
45
+ "22": "Seafood",
46
+ "23": "Tree Nuts & Seeds",
47
+ "24": "Turkey, Other Poultry",
48
+ "25": "Vegetables",
49
+ "26": "Yogurt",
50
+ "3": "Butter",
51
+ "4": "Cheese",
52
+ "5": "Chicken",
53
+ "6": "Condiments & Snacks",
54
+ "7": "Eggs",
55
+ "8": "Fish (Farm-Raised)",
56
+ "9": "Fish (Wild)"
57
+ }
58
+ ],
59
+ [
60
+ "Level of Processing",
61
+ {
62
+ "0": "Culinary Ingredient",
63
+ "1": "Freshly Prepared",
64
+ "2": "Moderately Processed",
65
+ "3": "Non-Food",
66
+ "4": "Ultra-Processed",
67
+ "5": "Whole/Minimally Processed"
68
+ }
69
+ ],
70
+ [
71
+ "Primary Food Product Category",
72
+ {
73
+ "0": "Beef",
74
+ "1": "Beverages",
75
+ "10": "Fruit",
76
+ "11": "Grain Products",
77
+ "12": "Legumes",
78
+ "13": "Meals",
79
+ "14": "Meat",
80
+ "15": "Milk",
81
+ "16": "Milk & Dairy",
82
+ "17": "Non-Food",
83
+ "18": "Pork",
84
+ "19": "Produce",
85
+ "2": "Bread, Grains & Legumes",
86
+ "20": "Rice",
87
+ "21": "Roots & Tubers",
88
+ "22": "Seafood",
89
+ "23": "Tree Nuts & Seeds",
90
+ "24": "Turkey, Other Poultry",
91
+ "25": "Vegetables",
92
+ "26": "Yogurt",
93
+ "3": "Butter",
94
+ "4": "Cheese",
95
+ "5": "Chicken",
96
+ "6": "Condiments & Snacks",
97
+ "7": "Eggs",
98
+ "8": "Fish (Farm-Raised)",
99
+ "9": "Fish (Wild)"
100
+ }
101
+ ],
102
+ [
103
+ "Food Product Group",
104
+ {
105
+ "0": "Beverages",
106
+ "1": "Bread, Grains & Legumes",
107
+ "2": "Condiments & Snacks",
108
+ "3": "Meals",
109
+ "4": "Meat",
110
+ "5": "Milk & Dairy",
111
+ "6": "Non-Food",
112
+ "7": "Produce",
113
+ "8": "Seafood"
114
+ }
115
+ ],
116
+ [
117
+ "Primary Food Product Group",
118
+ {
119
+ "0": "Beverages",
120
+ "1": "Bread, Grains & Legumes",
121
+ "2": "Condiments & Snacks",
122
+ "3": "Meals",
123
+ "4": "Meat",
124
+ "5": "Milk & Dairy",
125
+ "6": "Non-Food",
126
+ "7": "Produce",
127
+ "8": "Seafood"
128
+ }
129
+ ]
130
+ ],
131
  "dim": 768,
132
  "dropout": 0.1,
133
  "hidden_dim": 3072,
 
137
  "n_heads": 12,
138
  "n_layers": 6,
139
  "num_categories_per_task": [
140
+ 27,
141
  6,
142
+ 27,
143
+ 9,
144
+ 9
145
  ],
146
  "pad_token_id": 0,
147
  "qa_dropout": 0.1,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:80b47cc29dca4cdfda2885d8982aaffdbca2800f173b8dfc175ded23805ac652
3
- size 270321146
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8161350aaa17d641ff635e04d8ee86c8833c4b9c5d0f6db1177155ae0aab71df
3
+ size 271518634