zouharvi commited on
Commit
d3e9b80
·
verified ·
1 Parent(s): 66fd504

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
  )