Added more entities to training set and also randomized wine label. Earlier training set was flawed because wine labels were always located at the beginning of the description of the wine and the model trained itself to tag everything in the begining of the description as the wine label.

config.json

@@ -11,42 +11,42 @@
   "hidden_size": 768,
   "id2label": {
     "0": "O",
-    "1": "B-classification",
+    "1": "B-adj",
     "2": "B-country",
     "3": "B-flavor",
-    "4": "B-mouthfeel",
-    "5": "B-producer",
-    "6": "B-province",
-    "7": "B-region",
-    "8": "B-subregion",
-    "9": "B-wine",
-    "10": "I-classification",
-    "11": "I-country",
-    "12": "I-producer",
-    "13": "I-province",
-    "14": "I-region",
-    "15": "I-subregion",
-    "16": "I-wine"
+    "4": "B-grape",
+    "5": "B-moutfeel",
+    "6": "B-mouthfeel",
+    "7": "B-producer",
+    "8": "B-province",
+    "9": "B-region",
+    "10": "B-sweetness",
+    "11": "B-vintage",
+    "12": "I-country",
+    "13": "I-grape",
+    "14": "I-producer",
+    "15": "I-province",
+    "16": "I-region"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "B-classification": 1,
+    "B-adj": 1,
     "B-country": 2,
     "B-flavor": 3,
-    "B-mouthfeel": 4,
-    "B-producer": 5,
-    "B-province": 6,
-    "B-region": 7,
-    "B-subregion": 8,
-    "B-wine": 9,
-    "I-classification": 10,
-    "I-country": 11,
-    "I-producer": 12,
-    "I-province": 13,
-    "I-region": 14,
-    "I-subregion": 15,
-    "I-wine": 16,
+    "B-grape": 4,
+    "B-moutfeel": 5,
+    "B-mouthfeel": 6,
+    "B-producer": 7,
+    "B-province": 8,
+    "B-region": 9,
+    "B-sweetness": 10,
+    "B-vintage": 11,
+    "I-country": 12,
+    "I-grape": 13,
+    "I-producer": 14,
+    "I-province": 15,
+    "I-region": 16,
     "O": 0
   },
   "layer_norm_eps": 1e-12,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86b9e1494dfe489cddf8c0e5628dafb2e898ab0e6a9f6e5054845fa605925d92
+oid sha256:6d56cb4b0c77244c1b92a832a98cb7cf5217d54039804ca63d8bdaa5ea7aec69
 size 435686633

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a31d8a636cb808f2773176be3842a638157281a49cf8d736ce8bc5143ab71326
+size 3963