cheesyFishes
commited on
Commit
•
274e253
1
Parent(s):
81bb72e
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,17 @@
|
|
1 |
-
---
|
2 |
-
license: mit
|
3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
---
|
4 |
+
|
5 |
+
This repo contains the code needed to run AIR-Bench using llamaindex.
|
6 |
+
|
7 |
+
In this repo, I implement a custom retriever that uses `text-embedding-ada-002` as the dense embedding model, bm25 for sparse embeddings, and the `QueryFusionRetriever()` to combine results from these two, as well as generating extra queries for retrieval.
|
8 |
+
|
9 |
+
## Usage
|
10 |
+
|
11 |
+
`pip install llama-index llama-index-retrievers-bm25`
|
12 |
+
|
13 |
+
`python ./run_airbench.py`
|
14 |
+
|
15 |
+
## Customization
|
16 |
+
|
17 |
+
Feel free to use this as a template to evaluate other llama-index retrieval pipelines! You just need to customize the setup in `create_retriever_fn()` with the setup that you want to test against.
|