orionweller commited on
Commit
4b5c971
·
verified ·
1 Parent(s): 29bcd66

Update BM25S model

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ corpus.jsonl filter=lfs diff=lfs merge=lfs -text
37
+ corpus.mmindex.json filter=lfs diff=lfs merge=lfs -text
38
+ vocab.index.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ tags:
4
+ - bm25
5
+ - bm25s
6
+ - retrieval
7
+ - search
8
+ - lexical
9
+ ---
10
+
11
+ # BM25S Index
12
+
13
+ This is a BM25S index created with the [`bm25s` library](https://github.com/xhluca/bm25s) (version `0.1.3`), an ultra-fast implementation of BM25. It can be used for lexical retrieval tasks.
14
+
15
+ [BM25S GitHub Repository](https://github.com/xhluca/bm25s)
16
+
17
+ ## Installation
18
+
19
+ You can install the `bm25s` library with `pip`:
20
+
21
+ ```bash
22
+ pip install "bm25s==0.1.3"
23
+
24
+ # Include extra dependencies like stemmer
25
+ pip install "bm25s[full]==0.1.3"
26
+
27
+ # For huggingface hub usage
28
+ pip install huggingface_hub
29
+ ```
30
+
31
+ ## Loading a `bm25s` index
32
+
33
+ You can use this index for information retrieval tasks. Here is an example:
34
+
35
+ ```python
36
+ import bm25s
37
+ from bm25s.hf import BM25HF
38
+
39
+ # Load the index
40
+ retriever = BM25HF.load_from_hub("orionweller/kilt-100w-bm25", revision="main")
41
+
42
+ # You can retrieve now
43
+ query = "a cat is a feline"
44
+ results = retriever.retrieve(query, k=3)
45
+ ```
46
+
47
+ ## Saving a `bm25s` index
48
+
49
+ You can save a `bm25s` index to the Hugging Face Hub. Here is an example:
50
+
51
+ ```python
52
+ import bm25s
53
+ from bm25s.hf import BM25HF
54
+
55
+ # Create a BM25 index and add documents
56
+ retriever = BM25HF()
57
+ corpus = [
58
+ "a cat is a feline and likes to purr",
59
+ "a dog is the human's best friend and loves to play",
60
+ "a bird is a beautiful animal that can fly",
61
+ "a fish is a creature that lives in water and swims",
62
+ ]
63
+ corpus_tokens = bm25s.tokenize(corpus)
64
+ retriever.index(corpus_tokens)
65
+
66
+ token = None # You can get a token from the Hugging Face website
67
+ retriever.save_to_hub("orionweller/kilt-100w-bm25", token=token)
68
+ ```
69
+
70
+
71
+ ## Stats
72
+
73
+ This dataset was created using the following data:
74
+
75
+ | Statistic | Value |
76
+ | --- | --- |
77
+ | Number of documents | 24853658 |
78
+ | Number of tokens | 1232790853 |
79
+ | Average tokens per document | 49.601988286794644 |
80
+
81
+ ## Parameters
82
+
83
+ The index was created with the following parameters:
84
+
85
+ | Parameter | Value |
86
+ | --- | --- |
87
+ | k1 | `1.5` |
88
+ | b | `0.75` |
89
+ | delta | `0.5` |
90
+ | method | `lucene` |
91
+ | idf method | `lucene` |
corpus.jsonl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9de4342e66d35e4c604c9188d833da24f7de3fb0e837f489b2b0a5dc9098c76
3
+ size 15202866031
corpus.mmindex.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ec30db6658386771dede619abe6a555f351ba3fbbfd81187b9d6d8a4afd9668
3
+ size 280103400
data.csc.index.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e15e2f300f211a5f2aa7fc6a075de211c3ec02706e7f589cd12c48fd8edffdbc
3
+ size 4931163540
indices.csc.index.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:75072b69d9c9d107882306fdfc80e0e9cf90a83f25e481b1019a5aadf627fb5b
3
+ size 4931163540
indptr.csc.index.npy ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:805d08ed7e1b3a75168aef338bcc6757c6d70b7a397563d8bafc5946310f66a9
3
+ size 21935564
params.index.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "k1": 1.5,
3
+ "b": 0.75,
4
+ "delta": 0.5,
5
+ "method": "lucene",
6
+ "idf_method": "lucene",
7
+ "dtype": "float32",
8
+ "int_dtype": "int32",
9
+ "num_docs": 24853658,
10
+ "version": "0.1.3"
11
+ }
vocab.index.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4a939a50258cf8615fca63c589f01dd517952c3ee5685901ca912c44ed85654
3
+ size 113605687