Adr740 commited on
Commit
e7494c5
1 Parent(s): 844cee8

Update rag_benchmark.py

Browse files
Files changed (1) hide show
  1. rag_benchmark.py +2 -2
rag_benchmark.py CHANGED
@@ -1,8 +1,8 @@
1
 
2
  import pandas as pd
3
  from temporal_augmented_retrival import get_answer as get_temporal_answer
4
- from retrieval_classic_retrieval import get_answer as get_rag_answer
5
- from retrieval_naive import get_answer as get_naive_answer
6
 
7
  path_to_csv = "../contenu_embedded_august2023_1.csv"
8
  path_to_raw = "stockerbot-export.csv"
 
1
 
2
  import pandas as pd
3
  from temporal_augmented_retrival import get_answer as get_temporal_answer
4
+ from rag import get_answer as get_rag_answer
5
+ from naive_rag import get_answer as get_naive_answer
6
 
7
  path_to_csv = "../contenu_embedded_august2023_1.csv"
8
  path_to_raw = "stockerbot-export.csv"