Update README.md

README.md

@@ -53,11 +53,11 @@ sequence_to_classify = "Angela Merkel ist eine Politikerin in Deutschland und Vo
 candidate_labels = ["politics", "economy", "entertainment", "environment"]
 classifier(sequence_to_classify, candidate_labels)
 #{'sequence': 'Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU',
-# 'labels': ['environment', 'entertainment', 'economy', 'politics'],
-# 'scores': [0.32117515802383423,
-#  0.29328057169914246,
-#  0.27646392583847046,
-#  0.10908035188913345]}
+# 'labels': ['politics', 'economy', 'environment', 'entertainment'],
+# 'scores': [0.5737097859382629,
+#  0.22160571813583374,
+#  0.121183380484581,
+#  0.08350115269422531]}
 ```
 
 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
 candidate_labels = ["politics", "economy", "entertainment", "environment"]
 classifier(sequence_to_classify, candidate_labels, multi_label=True)
 #{'sequence': 'Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU',
-# 'labels': ['economy', 'politics', 'environment', 'entertainment'],
-# 'scores': [0.7499949932098389,
-#  0.710322380065918,
-#  0.5980390310287476,
-#  0.5272448658943176]}
+# 'labels': ['politics', 'economy', 'environment', 'entertainment'],
+# 'scores': [0.8171144723892212,
+#  0.4487208425998688,
+#  0.11949186027050018,
+#  0.04112538695335388]}
 ```
 
 ### Eval results
@@ -78,7 +78,7 @@ The model was evaluated using the XNLI test sets on 14 languages: English (en),
 
 |Datasets|en|ar|bg|de|el|es|fr|ru|sw|th|tr|ur|vi|zh|
 | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: | :---: |
-|[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|
+|[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|
 
 ### Training hyperparameters
 
@@ -91,7 +91,7 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_ratio: 0.1
-- num_epochs: 2
+- num_epochs: 1
 
 ### Framework versions
 - Transformers 4.28.1