panigrah commited on
Commit
9adf6e1
1 Parent(s): 4dfcc18

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -9
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
- "1": "classification",
50
- "2": "country",
51
- "3": "flavor",
52
- "4": "mouthfeel",
53
- "5": "producer",
54
- "6": "province",
55
- "7": "region",
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