Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,10 @@ widget:
|
|
| 18 |
|
| 19 |
Model trained from [roberta-base](https://huggingface.co/roberta-base) on the [go_emotions](https://huggingface.co/datasets/go_emotions) dataset for multi-label classification.
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
#### Dataset used for the model
|
| 22 |
|
| 23 |
[go_emotions](https://huggingface.co/datasets/go_emotions) is based on Reddit data and has 28 labels. It is a multi-label dataset where one or multiple labels may apply for any given input text, hence this model is a multi-label classification model with 28 'probability' float outputs for any given input text. Typically a threshold of 0.5 is applied to the probabilities for the prediction for each label.
|
|
|
|
| 18 |
|
| 19 |
Model trained from [roberta-base](https://huggingface.co/roberta-base) on the [go_emotions](https://huggingface.co/datasets/go_emotions) dataset for multi-label classification.
|
| 20 |
|
| 21 |
+
#### ONNX
|
| 22 |
+
|
| 23 |
+
An version of this model in ONNX format (including an INT8 quantized ONNX version) is now available at [https://huggingface.co/SamLowe/roberta-base-go_emotions-onnx](https://huggingface.co/SamLowe/roberta-base-go_emotions-onnx)
|
| 24 |
+
|
| 25 |
#### Dataset used for the model
|
| 26 |
|
| 27 |
[go_emotions](https://huggingface.co/datasets/go_emotions) is based on Reddit data and has 28 labels. It is a multi-label dataset where one or multiple labels may apply for any given input text, hence this model is a multi-label classification model with 28 'probability' float outputs for any given input text. Typically a threshold of 0.5 is applied to the probabilities for the prediction for each label.
|