Update README.md

README.md

@@ -101,7 +101,7 @@ base_model:
 - FacebookAI/xlm-roberta-large
 ---
 
-# COMET-poly-ic1-wmt25
+# COMET-poly-ic3-wmt25
 
 This model is based on [COMET-poly](https://github.com/zouharvi/COMET-poly), which is a fork but not compatible with original Unbabel's COMET.
 To run the model, you need to first install this version of COMET either with:
@@ -115,38 +115,57 @@ cd COMET-poly
 pip3 install -e comet_poly
 ```
 
-This model scores the translation `mt` but takes additional in-context example: source `src2`, translation `mt2`, and score `score2`, which makes it a better quality estimator:
+This model scores the translation `mt` but takes additional three in-context example: sources `src2`, `src3`, `src4`, translations `mt2`, `mt3`, `mt4`, and scores `score2`, `score3`, `score4`, which makes it a better quality estimator:
 ```python
 import comet_poly
-model = comet_poly.load_from_checkpoint(comet_poly.download_model("zouharvi/COMET-poly-ic1-wmt25"))
+model = comet_poly.load_from_checkpoint(comet_poly.download_model("zouharvi/COMET-poly-ic3-wmt25"))
 data = [
     {
         "src": "Iceberg lettuce got its name in the 1920s when it was shipped packed in ice to stay fresh.",
         "mt": "Eisbergsalat erhielt seinen Namen in den 1920er-Jahren, als er in Eis verpackt verschickt wurde, um frisch zu bleiben.",
         "src2": "Lettuce is mostly water, which helps keep it crisp when chilled.",
         "mt2": "Kopfsalat besteht größtenteils aus Wasser, was ihm hilft, beim Kühlen knackig zu bleiben.",
-        "score2": 94.5
+        "score2": 94.5,
+        "src3": "Iceberg lettuce is often used in burgers for its crunch and mild flavor.",
+        "mt3": "Íssalat er oft notað í hamborgara vegna stökkleika og milds bragðs.",
+        "score3": 92.0,
+        "src4": "Farmers harvest lettuce early in the morning to keep it fresh longer.",
+        "mt4": "Les agriculteurs récoltent la laitue tôt le matin pour la garder fraîche plus longtemps.",
+        "score4": 82.3
     },
     {
         "src": "Goats have rectangular pupils, which give them a wide field of vision—up to 320 degrees!",
         "mt": "Kozy mají obdélníkové zornice, což jim umožňuje vidět skoro všude kolem sebe, aniž by musely otáčet hlavou.",
         "src2": "Sheep, like goats, also have rectangular pupils for better peripheral vision.",
         "mt2": "Вівці, як і кози, також мають прямокутні зіниці для кращого периферичного зору.",
-        "score2": 96.0
+        "score2": 96.0,
+        "src3": "This unique eye shape helps them detect predators early.",
+        "mt3": "Ця унікальна форма очей допомагає їм рано виявляти хижаків.",
+        "score3": 93.2,
+        "src4": "Goats' vision stays stable even when they climb steep surfaces.",
+        "mt4": "Kozí vidění zůstává stabilní když jdou na vysoký svah.",
+        "score4": 67.4
     },
     {
         "src": "This helps them spot predators from almost all directions without moving their heads.",
         "mt": "Điều này giúp chúng phát hiện kẻ săn mồi từ gần như mọi hướng mà không cần quay đầu.",
         "src2": "Many prey animals have evolved to detect threats with minimal movement.",
         "mt2": "Nhiều động vật thịt có tiến hóa để xem mối nguy bằng nhỏ đi lại.",
-        "score2": 42.3
+        "score2": 42.3,
+        "src3": "Nepohybující se držení těla pomáhá zvířatům zůstat bez povšimnutí a přitom být ostražitá.",
+        "mt3": "Нерухома поза допомагає тваринам залишатися непоміченими, залишаючись при цьому пильними.",
+        "score3": 90.1,
+        "src4": "Animals like deer rely on peripheral vision to detect danger.",
+        "mt4": "Los animales como los ciervos confían en la visión periférica para detectar el peligro.",
+        "score4": 85.0
     }
 ]
+
 print("scores", model.predict(data, batch_size=8, gpus=1).scores)
 ```
 Outputs:
 ```
-scores [98.09857940673828, 85.52458953857422, 83.38972473144531]
+scores [98.42459869384766, 94.70307922363281, 91.14827728271484]
 ```
 
 You can use a readily-available training data to do the on-the-fly retrieval.
@@ -173,7 +192,7 @@ data_kb = list(datasets.load_dataset("zouharvi/wmt-human-all", split="train"))
 data_retrieved = comet_poly.retrieval.retrieve_from_kb(
     data=data,
     data_kb=data_kb,
-    k=1,
+    k=3, # this model takes three in-context examples
     prevent_hardmatch=False,
     key="src",
 )