MuntasirHossain
commited on
Commit
•
2ac566b
1
Parent(s):
d0abe7b
Update README.md
Browse files
README.md
CHANGED
@@ -52,15 +52,16 @@ It achieves the following results on the evaluation set:
|
|
52 |
|
53 |
## Model description
|
54 |
|
55 |
-
|
|
|
56 |
|
57 |
## Intended uses & limitations
|
58 |
|
59 |
-
|
60 |
|
61 |
## Training and evaluation data
|
62 |
|
63 |
-
|
64 |
|
65 |
## Training procedure
|
66 |
|
|
|
52 |
|
53 |
## Model description
|
54 |
|
55 |
+
This is a RoBERTa model fine-tuned on the [emotion](https://huggingface.co/datasets/dair-ai/emotion) to determine whether a text is within any of the six categories:
|
56 |
+
'sadness', 'joy', 'love', 'anger', 'fear', 'surprise'. The Trainer API was used to train the model.
|
57 |
|
58 |
## Intended uses & limitations
|
59 |
|
60 |
+
|
61 |
|
62 |
## Training and evaluation data
|
63 |
|
64 |
+
🤗 ``load_dataset`` package was used to load the data from the hub.
|
65 |
|
66 |
## Training procedure
|
67 |
|