updating hyperparams for knn+neighborsampling, necesarry to get the g…

…olden triples in our retrieved subgraph before filtering with PCST
pyg-team · Jan 9, 2025 · b1ff4b7 · b1ff4b7
1 parent 857d35a
commit b1ff4b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/llm/hotpot_qa.py b/examples/llm/hotpot_qa.py
@@ -115,7 +115,7 @@
     # k for KNN
     knn_neighsample_bs = 4096
     # number of neighbors for each seed node selected by KNN
-    fanout = 200
+    fanout = 100
     # number of hops for neighborsampling
     num_hops = 2
     query_loader = RAGQueryLoader(