bsingh commited on
Commit
5c7c36e
1 Parent(s): af28057

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -19,8 +19,7 @@ widget:
19
  - The training script is provided here: https://github.com/bsinghpratap/roberta_train_goEmotion
20
  - The model works well on most of the emotions except: 'desire', 'disgust', 'embarrassment', 'excitement', 'fear', 'grief', 'nervousness', 'pride', 'relief', 'remorse', 'surprise']
21
  - I'll try to fine-tune the model further and update here if RoBERTa achieves a better performance.
22
- - Each text datapoint can have more than 1 label. Most of the training set had 1 label: Counter({1: 36308, 2: 6541, 3: 532, 4: 28, 5: 1})
23
- -- So currently I just used the first label for each of the datapoint. Not ideal but it does a decent job.
24
 
25
  ## Model Performance
26
  ============================================================<br>
 
19
  - The training script is provided here: https://github.com/bsinghpratap/roberta_train_goEmotion
20
  - The model works well on most of the emotions except: 'desire', 'disgust', 'embarrassment', 'excitement', 'fear', 'grief', 'nervousness', 'pride', 'relief', 'remorse', 'surprise']
21
  - I'll try to fine-tune the model further and update here if RoBERTa achieves a better performance.
22
+ - Each text datapoint can have more than 1 label. Most of the training set had 1 label: Counter({1: 36308, 2: 6541, 3: 532, 4: 28, 5: 1}). So currently I just used the first label for each of the datapoint. Not ideal but it does a decent job.
 
23
 
24
  ## Model Performance
25
  ============================================================<br>