Commit
•
e9f7377
1
Parent(s):
9e178bb
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -3,7 +3,7 @@
|
|
3 |
---
|
4 |
license: mit
|
5 |
language:
|
6 |
-
-
|
7 |
tags:
|
8 |
- zero-shot-classification
|
9 |
- text-classification
|
@@ -14,7 +14,7 @@ metrics:
|
|
14 |
---
|
15 |
# xlm-roberta-large-french-party-cap-v3
|
16 |
## Model description
|
17 |
-
An `xlm-roberta-large` model finetuned on
|
18 |
|
19 |
## How to use the model
|
20 |
#### Loading and tokenizing input data
|
@@ -47,8 +47,10 @@ dataset = hg_data.map(tokenize_dataset, batched=True, remove_columns=hg_data.col
|
|
47 |
#### Inference using the Trainer class
|
48 |
```python
|
49 |
model = AutoModelForSequenceClassification.from_pretrained('poltextlab/xlm-roberta-large-french-party-cap-v3',
|
50 |
-
num_labels=
|
51 |
-
problem_type="multi_label_classification"
|
|
|
|
|
52 |
|
53 |
training_args = TrainingArguments(
|
54 |
output_dir='.',
|
|
|
3 |
---
|
4 |
license: mit
|
5 |
language:
|
6 |
+
- multilingual
|
7 |
tags:
|
8 |
- zero-shot-classification
|
9 |
- text-classification
|
|
|
14 |
---
|
15 |
# xlm-roberta-large-french-party-cap-v3
|
16 |
## Model description
|
17 |
+
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/).
|
18 |
|
19 |
## How to use the model
|
20 |
#### Loading and tokenizing input data
|
|
|
47 |
#### Inference using the Trainer class
|
48 |
```python
|
49 |
model = AutoModelForSequenceClassification.from_pretrained('poltextlab/xlm-roberta-large-french-party-cap-v3',
|
50 |
+
num_labels=num_labels,
|
51 |
+
problem_type="multi_label_classification",
|
52 |
+
ignore_mismatched_sizes=True
|
53 |
+
)
|
54 |
|
55 |
training_args = TrainingArguments(
|
56 |
output_dir='.',
|