--- --- language: - da tags: - text-classification - pytorch metrics: - accuracy - f1-score --- # xlm-roberta-large-danish-execspeech-cap-v3 ## Model description An `xlm-roberta-large` model fine-tuned on danish training data containing executive speeches labeled 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 This snippet prints the three most probable labels and their corresponding softmax scores: ```python import torch from transformers import AutoModelForSequenceClassification, AutoTokenizer model = AutoModelForSequenceClassification.from_pretrained("poltextlab/xlm-roberta-large-danish-execspeech-cap-v3") tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-large") sentence = "This is an example." inputs = tokenizer(sentence, return_tensors="pt", max_length=512, padding="do_not_pad", truncation=True ) logits = model(**inputs).logits probs = torch.softmax(logits, dim=1).tolist()[0] probs = {model.config.id2label[index]: round(probability, 2) for index, probability in enumerate(probs)} top3_probs = dict(sorted(probs.items(), key=lambda item: item[1], reverse=True)[:3]) print(top3_probs) ``` ## Model performance The model was evaluated on a test set of 1152 examples.
Model accuracy is **0.64**. | label | precision | recall | f1-score | support | |:-------------|------------:|---------:|-----------:|----------:| | 0 | 0.64 | 0.68 | 0.66 | 182 | | 1 | 0.33 | 0.18 | 0.24 | 11 | | 2 | 0.82 | 0.76 | 0.79 | 99 | | 3 | 0 | 0 | 0 | 5 | | 4 | 0.53 | 0.5 | 0.51 | 74 | | 5 | 0.68 | 0.79 | 0.73 | 136 | | 6 | 0.63 | 0.57 | 0.6 | 21 | | 7 | 0.55 | 0.87 | 0.68 | 30 | | 8 | 0.67 | 0.65 | 0.66 | 62 | | 9 | 1 | 0.77 | 0.87 | 13 | | 10 | 0.53 | 0.47 | 0.5 | 36 | | 11 | 0.41 | 0.52 | 0.45 | 58 | | 12 | 0.68 | 0.56 | 0.61 | 27 | | 13 | 0.55 | 0.45 | 0.49 | 40 | | 14 | 0.72 | 0.61 | 0.66 | 51 | | 15 | 0 | 0 | 0 | 5 | | 16 | 0.78 | 0.23 | 0.35 | 31 | | 17 | 0.59 | 0.69 | 0.64 | 118 | | 18 | 0.61 | 0.58 | 0.59 | 103 | | 19 | 0.79 | 0.78 | 0.78 | 49 | | 20 | 0 | 0 | 0 | 1 | | macro avg | 0.55 | 0.51 | 0.52 | 1152 | | weighted avg | 0.64 | 0.64 | 0.63 | 1152 | ### Fine-tuning procedure This model was fine-tuned with the following key hyperparameters: - **Number of Training Epochs**: 10 - **Batch Size**: 8 - **Learning Rate**: 5e-06 - **Early Stopping**: enabled with a patience of 2 epochs ## 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). ## Reference 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 ## 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.