Update README.md
Browse files
README.md
CHANGED
@@ -173,7 +173,7 @@ data_kb = list(datasets.load_dataset("zouharvi/wmt-human-all", split="train"))
|
|
173 |
data_retrieved = comet_poly.retrieval.retrieve_from_kb(
|
174 |
data=data,
|
175 |
data_kb=data_kb,
|
176 |
-
k=1,
|
177 |
prevent_hardmatch=False,
|
178 |
key="src",
|
179 |
)
|
|
|
173 |
data_retrieved = comet_poly.retrieval.retrieve_from_kb(
|
174 |
data=data,
|
175 |
data_kb=data_kb,
|
176 |
+
k=1, # this model takes one in-context example
|
177 |
prevent_hardmatch=False,
|
178 |
key="src",
|
179 |
)
|