MORESPulse / README.md
ringorsolya's picture
Upload folder using huggingface_hub
e1e2019 verified
|
Raw
History Blame
3.59 kB
---
license: cc-by-4.0
language:
- multilingual
tags:
- text-classification
- pytorch
metrics:
- f1-score
extra_gated_fields:
Country: country
Institution: text
Institution Email: text
Full Name: text
Please specify your academic project/use case you want to use the models for: text
extra_gated_prompt: Our models are intended for academic projects and academic research
only. If you are not affiliated with an academic institution, please reach out to
us at huggingface [at] poltextlab [dot] com for further inquiry. If we cannot clearly
determine your academic affiliation and use case based on your form data, your request
may be rejected. Please allow us a few business days to manually review subscriptions.
---
# xlm-roberta-large-pooled-MORES
## Model description
An `xlm-roberta-large` model finetuned on sentence-level multilingual training data hand-annotated using the following labels:
- **0**: "Anger"
- **1**: "Fear"
- **2**: "Disgust"
- **3**: "Sadness"
- **4**: "Joy"
- **5**: "None of Them"
This model can also be used for sentiment classification with the following conversion:
- **Joy (4)** → Positive
- **None of Them (5)** → Neutral (or None of Them)
- **All Other Labels** → Negative
The training data we used was augmented using artificially generated examples and translated texts. It covers 7 languages (English, German, French, Polish, Slovak, Czech and Hungarian) with nearly identical shares.
## How to use the model
```python
from transformers import AutoTokenizer, pipeline
tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-large")
pipe = pipeline(
model="poltextlab/xlm-roberta-large-pooled-MORES",
task="text-classification",
tokenizer=tokenizer,
use_fast=False,
token="<your_hf_read_only_token>"
)
text = "We will place an immediate 6-month halt on the finance driven closure of beds and wards, and set up an independent audit of needs and facilities."
pipe(text)
```
### Gated access
Due to the gated access, you must pass the `token` parameter when loading the model. In earlier versions of the Transformers package, you may need to use the `use_auth_token` parameter instead.
## Model performance
The model was evaluated on language-specific test sets and demonstrated nearly identical performance across all languages:
![Model benchmark (language-specific test)](v5_fixed_f1_scores.png)
### Fine-tuning procedure
This model was fine-tuned with the following key hyperparameters:
- **Number of Training Epochs**: 10
- **Batch Size**: 16
- **Learning Rate**: 5e-06
- **Early Stopping**: enabled with a patience of 2 epochs
## Inference platform
This model is used by the [Babel Machine](https://babel.poltextlab.com), an open-source and free natural language processing tool, designed to simplify and speed up projects for comparative research.
## Cooperation
Model performance can be significantly improved by extending our training sets. We appreciate every submission of coded corpora (of any domain and language) at poltextlab{at}poltextlab{dot}com or by using the [Babel Machine](https://babel.poltextlab.com).
## Debugging and issues
This architecture uses the `sentencepiece` tokenizer. In order to use the model before `transformers==4.27` you need to install it manually.
If you encounter a `RuntimeError` when loading the model using the `from_pretrained()` method, adding `ignore_mismatched_sizes=True` should solve the issue.
## Funding
The research was funded by European Union’s Horizon 2020 research and innovation program, “MORES” project (Grant No.: 101132601).