allen-li1231
/

treehop-rag

Information Retrieval

Retrieval-Augmented Generation

model_hub_mixin

multi-hop question answering

pytorch_model_hub_mixin

Model card Files Files and versions Community

allen-li1231 commited on 17 days ago

Commit

d8651f0

·

verified ·

1 Parent(s): 4ae4690

Update README.md

Add link to repository.

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -15,6 +15,7 @@ tags:
 # TreeHop: Generate and Filter Next Query Embeddings Efficiently for Multi-hop Question Answering
 [![arXiv](https://img.shields.io/badge/arXiv-2504.20114-b31b1b.svg?style=flat)](https://arxiv.org/abs/2504.20114)
 [![HuggingFace](https://img.shields.io/badge/HuggingFace-Model-blue.svg)](https://huggingface.co/allen-li1231/treehop-rag)
 [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://img.shields.io/badge/license-MIT-blue)
@@ -59,7 +60,7 @@ Please refer to [requirements.txt](/requirements.txt).
 ## Preliminaries
-This repository comes with [evaluate embedding databases](./embedding_data/) for reproduction purpose. Activate [git LFS](https://docs.github.com/en/repositories/working-with-files/managing-large-files/installing-git-large-file-storage) to clone the repository using `git lfs clone [LINK_TO_REPO]`, or pull the data under the existing local repository using:
 ```sh
 git lfs pull
 ```
@@ -77,7 +78,7 @@ python init_multihop_rag.py
 ## Multi-hop Retrieval with TreeHop: How-to-Use
 Here we use [MultiHop RAG evaluate dataset](https://arxiv.org/abs/2401.15391) in the example below.
-The repository comes with the necessary files for the example to run, see [preliminaries](#preliminaries).
 ```python
 from tree_hop import TreeHopModel

 # TreeHop: Generate and Filter Next Query Embeddings Efficiently for Multi-hop Question Answering
+[![GitHub](https://img.shields.io/badge/GitHub-%23121011.svg?logo=github&logoColor=white)](https://github.com/allen-li1231/TreeHop-RAG)
 [![arXiv](https://img.shields.io/badge/arXiv-2504.20114-b31b1b.svg?style=flat)](https://arxiv.org/abs/2504.20114)
 [![HuggingFace](https://img.shields.io/badge/HuggingFace-Model-blue.svg)](https://huggingface.co/allen-li1231/treehop-rag)
 [![License](https://img.shields.io/badge/License-MIT-blue.svg)](https://img.shields.io/badge/license-MIT-blue)
 ## Preliminaries
+[This repository](https://github.com/allen-li1231/TreeHop-RAG) comes with [evaluate embedding databases](./embedding_data/) for reproduction purpose. Activate [git LFS](https://docs.github.com/en/repositories/working-with-files/managing-large-files/installing-git-large-file-storage) to clone the repository using `git lfs clone [LINK_TO_REPO]`, or pull the data under the existing local repository using:
 ```sh
 git lfs pull
 ```
 ## Multi-hop Retrieval with TreeHop: How-to-Use
 Here we use [MultiHop RAG evaluate dataset](https://arxiv.org/abs/2401.15391) in the example below.
+[The repository](https://github.com/allen-li1231/TreeHop-RAG) comes with the necessary files for the example to run, see [preliminaries](#preliminaries).
 ```python
 from tree_hop import TreeHopModel