Skip to content

kumo-ai/ContextGNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

b86641f · Aug 28, 2024
Jul 20, 2024
Aug 28, 2024
Aug 18, 2024
Aug 28, 2024
Aug 20, 2024
Aug 8, 2024
Jul 20, 2024
Jul 4, 2024
Aug 13, 2024
Jul 31, 2024

Repository files navigation

hybridgnn

How to Run

Run benchmark/relbench_link_prediction_benchmark.py

python relbench_link_prediction_benchmark.py --dataset rel-stack --task user-post-comment --model rhstransformer --num_trials 10 
python relbench_link_prediction_benchmark.py --dataset rel-hm --task user-item-purcahse --model rhstransformer
python relbench_link_prediction_benchmark.py --dataset rel-trial --task site-sponsor-run --model hybridgnn --num_layers 4

Run examples/relbench_example.py

python relbench_example.py --dataset rel-trial --task site-sponsor-run --model hybridgnn --num_layers 4
python relbench_example.py --dataset rel-trial --task condition-sponsor-run --model hybridgnn --num_layers 4

Install Instruction

pip install -e .

# to run examples and benchmarks
pip install -e '.[full]'

pip install -U sentence-transformers