Commit
·
e610672
1
Parent(s):
f7f9bd0
Update README.md
Browse files
README.md
CHANGED
|
@@ -5,7 +5,7 @@ metrics:
|
|
| 5 |
- f1
|
| 6 |
pipeline_tag: text-classification
|
| 7 |
datasets:
|
| 8 |
-
-
|
| 9 |
license: apache-2.0
|
| 10 |
---
|
| 11 |
|
|
@@ -20,7 +20,7 @@ This model is a fine-tuned version of [PlanTL-GOB-ES/roberta-base-bne](https://h
|
|
| 20 |
>>> from transformers import pipeline
|
| 21 |
|
| 22 |
|
| 23 |
-
>>> model_name= '
|
| 24 |
>>> pipe = pipeline("text-classification", model=model_name)
|
| 25 |
|
| 26 |
>>> pipe("Quiero acabar con todo. No merece la pena vivir.")
|
|
@@ -39,7 +39,7 @@ The dataset consists of comments on Reddit, Twitter, and inputs/outputs of the A
|
|
| 39 |
|
| 40 |
The dataset has 10050 rows (777 considered as Suicidal Ideation/Behavior and 9273 considered Non-Suicidal).
|
| 41 |
|
| 42 |
-
More info: https://huggingface.co/datasets/
|
| 43 |
|
| 44 |
## Training procedure
|
| 45 |
|
|
|
|
| 5 |
- f1
|
| 6 |
pipeline_tag: text-classification
|
| 7 |
datasets:
|
| 8 |
+
- hackathon-somos-nlp-2023/suicide-comments-es
|
| 9 |
license: apache-2.0
|
| 10 |
---
|
| 11 |
|
|
|
|
| 20 |
>>> from transformers import pipeline
|
| 21 |
|
| 22 |
|
| 23 |
+
>>> model_name= 'hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es'
|
| 24 |
>>> pipe = pipeline("text-classification", model=model_name)
|
| 25 |
|
| 26 |
>>> pipe("Quiero acabar con todo. No merece la pena vivir.")
|
|
|
|
| 39 |
|
| 40 |
The dataset has 10050 rows (777 considered as Suicidal Ideation/Behavior and 9273 considered Non-Suicidal).
|
| 41 |
|
| 42 |
+
More info: https://huggingface.co/datasets/hackathon-somos-nlp-2023/suicide-comments-es
|
| 43 |
|
| 44 |
## Training procedure
|
| 45 |
|