antoinelouis commited on
Commit
a280fde
1 Parent(s): 97d37d7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -2
README.md CHANGED
@@ -21,7 +21,7 @@ This is a [ColBERTv1](https://github.com/stanford-futuredata/ColBERT) model: it
21
 
22
  To use this model, you will need to install the following libraries:
23
  ```
24
- pip install colbert-ir[faiss-gpu] faiss torch
25
  ```
26
 
27
 
@@ -67,7 +67,12 @@ with Run().context(RunConfig(nranks=n_gpu,experiment=experiment)):
67
 
68
  ## Evaluation
69
 
70
- *(tba)*
 
 
 
 
 
71
 
72
  ## Training
73
 
 
21
 
22
  To use this model, you will need to install the following libraries:
23
  ```
24
+ pip install colbert-ir @ git+https://github.com/stanford-futuredata/ColBERT.git faiss-gpu==1.7.2
25
  ```
26
 
27
 
 
67
 
68
  ## Evaluation
69
 
70
+ We evaluated our model on the smaller development set of mMARCO-fr, which consists of 6,980 queries for a corpus of 8.8M candidate passages. Below, we compared the model performance with a biencoder model fine-tuned on the same dataset. We report the mean reciprocal rank (MRR) and recall at various cut-offs (R@k).
71
+
72
+ | model | Vocab. | #Param. | Size | MRR@10 | R@10 | R@100(↑) | R@500 |
73
+ |:------------------------------------------------------------------------------------------------------------------------|:-------|--------:|------:|---------:|-------:|-----------:|--------:|
74
+ | **colbertv1-camembert-base-mmarcoFR** | 🇫🇷 | 110M | 443MB | 29.51 | 54.21 | 80.00 | 88.40 |
75
+ | [biencoder-camembert-base-mmarcoFR](https://huggingface.co/antoinelouis/biencoder-camembert-base-mmarcoFR) | 🇫🇷 | 110M | 438MB | 28.53 | 51.46 | 77.82 | 89.13 |
76
 
77
  ## Training
78