mjwong commited on
Commit
a07f1ba
1 Parent(s): 2636d55

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -12
README.md CHANGED
@@ -53,11 +53,11 @@ sequence_to_classify = "Angela Merkel ist eine Politikerin in Deutschland und Vo
53
  candidate_labels = ["politics", "economy", "entertainment", "environment"]
54
  classifier(sequence_to_classify, candidate_labels)
55
  #{'sequence': 'Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU',
56
- # 'labels': ['environment', 'entertainment', 'economy', 'politics'],
57
- # 'scores': [0.32117515802383423,
58
- # 0.29328057169914246,
59
- # 0.27646392583847046,
60
- # 0.10908035188913345]}
61
  ```
62
 
63
  If more than one candidate label can be correct, pass `multi_class=True` to calculate each class independently:
@@ -66,11 +66,11 @@ If more than one candidate label can be correct, pass `multi_class=True` to calc
66
  candidate_labels = ["politics", "economy", "entertainment", "environment"]
67
  classifier(sequence_to_classify, candidate_labels, multi_label=True)
68
  #{'sequence': 'Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU',
69
- # 'labels': ['economy', 'politics', 'environment', 'entertainment'],
70
- # 'scores': [0.7499949932098389,
71
- # 0.710322380065918,
72
- # 0.5980390310287476,
73
- # 0.5272448658943176]}
74
  ```
75
 
76
  ### Eval results
@@ -78,7 +78,7 @@ The model was evaluated using the XNLI test sets on 14 languages: English (en),
78
 
79
  |Datasets|en|ar|bg|de|el|es|fr|ru|sw|th|tr|ur|vi|zh|
80
  | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
81
- |[multilingual-e5-base-xnli](https://huggingface.co/mjwong/multilingual-e5-base-xnli)|0.835|0.758|0.792|0.788|0.788|0.803|0.795|0.773|0.719|0.754|0.757|0.696|0.779|0.778|
82
 
83
  ### Training hyperparameters
84
 
@@ -91,7 +91,7 @@ The following hyperparameters were used during training:
91
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
92
  - lr_scheduler_type: linear
93
  - lr_scheduler_warmup_ratio: 0.1
94
- - num_epochs: 2
95
 
96
  ### Framework versions
97
  - Transformers 4.28.1
 
53
  candidate_labels = ["politics", "economy", "entertainment", "environment"]
54
  classifier(sequence_to_classify, candidate_labels)
55
  #{'sequence': 'Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU',
56
+ # 'labels': ['politics', 'economy', 'environment', 'entertainment'],
57
+ # 'scores': [0.5737097859382629,
58
+ # 0.22160571813583374,
59
+ # 0.121183380484581,
60
+ # 0.08350115269422531]}
61
  ```
62
 
63
  If more than one candidate label can be correct, pass `multi_class=True` to calculate each class independently:
 
66
  candidate_labels = ["politics", "economy", "entertainment", "environment"]
67
  classifier(sequence_to_classify, candidate_labels, multi_label=True)
68
  #{'sequence': 'Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU',
69
+ # 'labels': ['politics', 'economy', 'environment', 'entertainment'],
70
+ # 'scores': [0.8171144723892212,
71
+ # 0.4487208425998688,
72
+ # 0.11949186027050018,
73
+ # 0.04112538695335388]}
74
  ```
75
 
76
  ### Eval results
 
78
 
79
  |Datasets|en|ar|bg|de|el|es|fr|ru|sw|th|tr|ur|vi|zh|
80
  | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
81
+ |[multilingual-e5-base-xnli](https://huggingface.co/mjwong/multilingual-e5-base-xnli)|0.849|0.771|0.800|0.796|0.795|0.812|0.801|0.783|0.731|0.767|0.771|0.710|0.789|0.786|
82
 
83
  ### Training hyperparameters
84
 
 
91
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
92
  - lr_scheduler_type: linear
93
  - lr_scheduler_warmup_ratio: 0.1
94
+ - num_epochs: 1
95
 
96
  ### Framework versions
97
  - Transformers 4.28.1