JuliaGast / TGB2

Temporal Graph Benchmark project repo
1 stars 0 forks source link

Add STHN method for heterogeneous temporal graphs #33

Closed erfanloghmani closed 3 months ago

erfanloghmani commented 4 months ago

Adding STHN method from: https://github.com/celi52/STHN/tree/main

To run:

  1. Install requirements. Also updated requirements.txt. The two new requirements are pybind11 and torchmetrics==0.11.0
  2. compile the sampler
    python sthn_sampler_setup.py build_ext --inplace
  3. run the example code
    python sthn.py

If the code runs correctly the output would end with

INFO: Test: Evaluation Setting: >>> ONE-VS-MANY <<< 
        Test: mrr: X
        Test: Elapsed Time (s):  Y
shenyangHuang commented 4 months ago

Thanks Erfan, can you also add a README in the code so we keep track of it after it is merged? Just copy paste the install instruction and anything else you want to add into a STHN_README.md