poltextlab
/

xlm-roberta-large-social-cap-v3

Text Classification

zero-shot-classification

Inference Endpoints

Model card Files Files and versions Community

poltextlab commited on Nov 3, 2023

Commit

fb895ca

•

1 Parent(s): b5bdad6

Upload README.md with huggingface_hub

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -47,7 +47,7 @@ dataset = hg_data.map(tokenize_dataset, batched=True, remove_columns=hg_data.col
 #### Inference using the Trainer class
 ```python
 model = AutoModelForSequenceClassification.from_pretrained('poltextlab/xlm-roberta-large-social-cap-v3',
-                                                           num_labels=num_labels,
                                                            problem_type="multi_label_classification",
                                                            ignore_mismatched_sizes=True
                                                            )

 #### Inference using the Trainer class
 ```python
 model = AutoModelForSequenceClassification.from_pretrained('poltextlab/xlm-roberta-large-social-cap-v3',
+                                                           num_labels=22,
                                                            problem_type="multi_label_classification",
                                                            ignore_mismatched_sizes=True
                                                            )