poltextlab
commited on
Commit
•
502ee5a
1
Parent(s):
ffb1fd0
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ 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-parlspeech-cap-v3',
|
50 |
-
num_labels=
|
51 |
problem_type="multi_label_classification",
|
52 |
ignore_mismatched_sizes=True
|
53 |
)
|
|
|
47 |
#### Inference using the Trainer class
|
48 |
```python
|
49 |
model = AutoModelForSequenceClassification.from_pretrained('poltextlab/xlm-roberta-large-parlspeech-cap-v3',
|
50 |
+
num_labels=22,
|
51 |
problem_type="multi_label_classification",
|
52 |
ignore_mismatched_sizes=True
|
53 |
)
|