Update README.md
Browse files
README.md
CHANGED
@@ -10,6 +10,9 @@ tags:
|
|
10 |
|
11 |
Pretrained model on on wine labels and descriptions for named entity recognition that uses bert-base-uncased as the base model.
|
12 |
|
|
|
|
|
|
|
13 |
## Model description
|
14 |
|
15 |
|
@@ -46,15 +49,13 @@ balanced: mouthfeel: 0.99992
|
|
46 |
The BERT model was trained on 20K reviews and wine labels derived from https://huggingface.co/datasets/james-burton/wine_reviews_all_text and manually annotated to capture the following tokens
|
47 |
|
48 |
```
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
"8": "subregion",
|
57 |
-
"9": "wine"
|
58 |
```
|
59 |
|
60 |
## Training procedure
|
|
|
10 |
|
11 |
Pretrained model on on wine labels and descriptions for named entity recognition that uses bert-base-uncased as the base model.
|
12 |
|
13 |
+
* Updated to remove bias on position of wine label in the training inputs.
|
14 |
+
* also updated to remove trying to get the wine classification. e.g. Grand Cru etc because training data is not reliable.
|
15 |
+
|
16 |
## Model description
|
17 |
|
18 |
|
|
|
49 |
The BERT model was trained on 20K reviews and wine labels derived from https://huggingface.co/datasets/james-burton/wine_reviews_all_text and manually annotated to capture the following tokens
|
50 |
|
51 |
```
|
52 |
+
adjective: nice, exciting, strong etc
|
53 |
+
country: countries specified in label or description
|
54 |
+
flavor: fruit, apple, toast, smoke etc
|
55 |
+
grape: Cab, Cabernet Sauvignon, etc
|
56 |
+
mouthfeel: lucious, smooth, textured, rough etc
|
57 |
+
producer: wine maker
|
58 |
+
province, region: province and region of wine - sometimes these get mixed up
|
|
|
|
|
59 |
```
|
60 |
|
61 |
## Training procedure
|