Update README.md

README.md

@@ -51,29 +51,29 @@ BioHiCL aligns:
 ```python
 from beir import util
 from beir.datasets.data_loader import GenericDataLoader
-from sentence_transformers import SentenceTransformer
+from beir.retrieval.models import SentenceBERT
 from beir.retrieval.search.dense import DenseRetrievalExactSearch
 from beir.retrieval.evaluation import EvaluateRetrieval
 
-# Dataset
+
+# 1. Download  load the SciFact dataset
 dataset = "scifact"
-url = "https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/scifact.zip"
+url = "https://public.ukp.informatik.tu-darmstadt.de/thakur/BEIR/datasets/" + dataset + ".zip"
 
 data_path = util.download_and_unzip(url, "datasets")
 corpus, queries, qrels = GenericDataLoader(data_path).load(split="test")
 
-# Model
 model_name = "LunaLan07/BioHiCL-large"
-model = SentenceTransformer(model_name)
+model = SentenceBERT(model_name)
 
-# Retrieval
 retriever = DenseRetrievalExactSearch(model, batch_size=16)
-results = retriever.search(corpus, queries, top_k=10, score_function="cos_sim")
 
-# Evaluation
-ndcg, _map, recall, precision = EvaluateRetrieval.evaluate(
-    qrels, results, k_values=[1, 3, 5, 10]
-)
+top_k = 10  # top 10 documents per query
+results = retriever.search(corpus, queries, top_k=top_k, score_function="cos_sim")
+
+k_values = [1, 3, 5, 10]
+ndcg, _map, recall, precision = EvaluateRetrieval.evaluate(qrels, results, k_values=k_values)
+
 
 ```