Skip to content

Commit a22d5d7

Browse files
committed
minor fix
1 parent 64c98ac commit a22d5d7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

finetune-embedding/rag.py

+1
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ def main(batch_size,train_data,data_dir, model_name, output_dir,min_chunk_size,
107107
item.update({"chunks":[chunk.to_dict() for chunk in chunks]})
108108
subsample_list.append(item)
109109

110+
#TODO: call ragas after each batch & write results to json
110111
ragas_scores = get_ragas_score(subsample_list)["context_precision"]
111112
_ = [item.update({"ragas_score":score}) for item, score in zip(subsample_list, ragas_scores)]
112113
with open(os.path.join(output_dir, "subset.json"),"w") as file:

0 commit comments

Comments
 (0)