Update README.md
Browse files
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>
|