Closed erfanloghmani closed 3 months ago
Adding STHN method from: https://github.com/celi52/STHN/tree/main
To run:
requirements.txt
pybind11
torchmetrics==0.11.0
python sthn_sampler_setup.py build_ext --inplace
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
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
STHN_README.md
Adding STHN method from: https://github.com/celi52/STHN/tree/main
To run:
requirements.txt
. The two new requirements arepybind11
andtorchmetrics==0.11.0
If the code runs correctly the output would end with