alihan/RAG_Techniques
1
0
Fork 0
mirror of https://github.com/NirDiamant/RAG_Techniques.git synced 2025-04-07 00:48:52 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix scores initialization

Browse Source
This commit is contained in:
rotbka
2025-02-19 22:27:10 +02:00
parent d100326db5
commit c4eb7c15e6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
all_rag_techniques/dartboard.ipynb
View File

@@ -400,10 +400,10 @@
" document_probabilities = lognorm(document_distances, sigma)\n",
" \n",
" # Initialize with most relevant document\n",
" selection_scores = []\n",
" \n",
" most_relevant_idx = np.argmax(query_probabilities)\n",
" selected_indices = np.array([most_relevant_idx])\n",
" \n",
" selection_scores = [1.0] # dummy score for the first document\n",
" # Get initial distances from the first selected document\n",
" max_distances = document_probabilities[most_relevant_idx]\n",
" \n",