poltextlab commited on
Commit
d8b9eab
1 Parent(s): 6dd4a4e

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -12,7 +12,7 @@ metrics:
12
  - accuracy
13
  - f1-score
14
  ---
15
- # poltextlab/xlm-roberta-large-hungarian-cap-v3
16
  ## Model description
17
  An `xlm-roberta-large` model finetuned on hungarian training data labelled with [major topic codes](https://www.comparativeagendas.net/pages/master-codebook) from the [Comparative Agendas Project](https://www.comparativeagendas.net/).
18
 
@@ -46,7 +46,7 @@ dataset = hg_data.map(tokenize_dataset, batched=True, remove_columns=hg_data.col
46
 
47
  #### Inference using the Trainer class
48
  ```python
49
- model = AutoModelForSequenceClassification.from_pretrained('poltextlab/poltextlab/xlm-roberta-large-hungarian-cap-v3',
50
  num_labels=22,
51
  problem_type="multi_label_classification",
52
  ignore_mismatched_sizes=True
@@ -70,7 +70,7 @@ predicted = pd.DataFrame(np.argmax(probs, axis=1)).replace({0: CAP_NUM_DICT}).re
70
  ```
71
 
72
  ### Fine-tuning procedure
73
- `poltextlab/xlm-roberta-large-hungarian-cap-v3` was fine-tuned using the Hugging Face Trainer class with the following hyperparameters:
74
  ```python
75
  training_args = TrainingArguments(
76
  output_dir=f"../model/{model_dir}/tmp/",
 
12
  - accuracy
13
  - f1-score
14
  ---
15
+ # xlm-roberta-large-hungarian-cap-v3
16
  ## Model description
17
  An `xlm-roberta-large` model finetuned on hungarian training data labelled with [major topic codes](https://www.comparativeagendas.net/pages/master-codebook) from the [Comparative Agendas Project](https://www.comparativeagendas.net/).
18
 
 
46
 
47
  #### Inference using the Trainer class
48
  ```python
49
+ model = AutoModelForSequenceClassification.from_pretrained('poltextlab/xlm-roberta-large-hungarian-cap-v3',
50
  num_labels=22,
51
  problem_type="multi_label_classification",
52
  ignore_mismatched_sizes=True
 
70
  ```
71
 
72
  ### Fine-tuning procedure
73
+ `xlm-roberta-large-hungarian-cap-v3` was fine-tuned using the Hugging Face Trainer class with the following hyperparameters:
74
  ```python
75
  training_args = TrainingArguments(
76
  output_dir=f"../model/{model_dir}/tmp/",