--- license: mit language: - multilingual tags: - zero-shot-classification - text-classification - pytorch metrics: - accuracy - f1-score extra_gated_prompt: 'Our models are intended for academic use only. If you are not affiliated with an academic institution, please provide a rationale for using our models. If you use our models for your work or research, please cite this paper: Sebők, M., Máté, Á., Ring, O., Kovács, V., & Lehoczki, R. (2024). Leveraging Open Large Language Models for Multilingual Policy Topic Classification: The Babel Machine Approach. Social Science Computer Review, 0(0). https://doi.org/10.1177/08944393241259434' extra_gated_fields: Name: text Country: country Institution: text E-mail: text Use case: text --- # xlm-roberta-large-english-party-cap-v3 ## Model description An `xlm-roberta-large` model finetuned on multilingual training data containing texts of the `party` domain labelled with [major topic codes](https://www.comparativeagendas.net/pages/master-codebook) from the [Comparative Agendas Project](https://www.comparativeagendas.net/). ## 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-english-party-cap-v3", task="text-classification", tokenizer=tokenizer, use_fast=False, 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 a test set of 13879 examples (10% of the available data).
Model accuracy is **0.73**. | label | precision | recall | f1-score | support | |:-------------|------------:|---------:|-----------:|----------:| | 0 | 0.69 | 0.73 | 0.71 | 1142 | | 1 | 0.68 | 0.7 | 0.69 | 705 | | 2 | 0.79 | 0.85 | 0.82 | 865 | | 3 | 0.79 | 0.77 | 0.78 | 362 | | 4 | 0.72 | 0.64 | 0.68 | 628 | | 5 | 0.87 | 0.83 | 0.85 | 936 | | 6 | 0.68 | 0.71 | 0.7 | 430 | | 7 | 0.88 | 0.8 | 0.84 | 360 | | 8 | 0.72 | 0.75 | 0.74 | 198 | | 9 | 0.85 | 0.79 | 0.82 | 327 | | 10 | 0.8 | 0.75 | 0.77 | 903 | | 11 | 0.61 | 0.68 | 0.64 | 752 | | 12 | 0.66 | 0.79 | 0.72 | 531 | | 13 | 0.65 | 0.61 | 0.63 | 406 | | 14 | 0.83 | 0.75 | 0.79 | 964 | | 15 | 0.71 | 0.74 | 0.73 | 234 | | 16 | 0.71 | 0.67 | 0.69 | 253 | | 17 | 0.77 | 0.83 | 0.8 | 1637 | | 18 | 0.71 | 0.59 | 0.65 | 910 | | 19 | 0.73 | 0.74 | 0.73 | 366 | | 20 | 0.76 | 0.61 | 0.68 | 77 | | 21 | 0.59 | 0.6 | 0.59 | 893 | | macro avg | 0.74 | 0.72 | 0.73 | 13879 | | weighted avg | 0.74 | 0.73 | 0.73 | 13879 | ## Inference platform This model is used by the [CAP 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 CAP-coded corpora (of any domain and language) at poltextlab{at}poltextlab{dot}com or by using the [CAP Babel Machine](https://babel.poltextlab.com). ## Debugging and issues This architecture uses the `sentencepiece` tokenizer. In order to run 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.