We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c4562a3 commit 0a99920Copy full SHA for 0a99920
torch_geometric/utils/rag/feature_store.py
@@ -14,6 +14,7 @@
14
from torch_geometric.typing import InputEdges, InputNodes
15
from torch_geometric.utils.rag.backend_utils import batch_knn
16
17
+
18
# NOTE: Only compatible with Homogeneous graphs for now
19
class KNNRAGFeatureStore(LocalFeatureStore):
20
"""A feature store that uses a KNN-based approach to retrieve seed nodes and edges.
0 commit comments