awslabs / awslabs/graphstorm

Use more efficient KNN methods for link prediction

Open
#522 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
452
Forks
76
PR merge metrics
No merged PRs in 30d

Description

We should implement a proper link prediction inference script. Users can provide a list of nodes and the link prediction inference script should return top K nodes that are likely to connect with the input nodes. We should use FAISS for link prediction.

Contributor guide

Open the contributing guide

Research direction

No file, test, or entry point is named in the issue, so first locate any existing inference scripts and determine how link prediction is represented. Define the FAISS-based inference flow around a supplied list of nodes. Done means the script accepts those nodes and returns the top K nodes most likely to connect to them.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.