Skip to content

GitLab

Unverified Commit 3ebf2660 authored by Theofilos Papapanagiotou's avatar Theofilos Papapanagiotou Committed by GitHub
Browse files

correct PinSAGE knn example syntax (#1617) 


Co-authored-by: default avatarQuan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: default avatarChao Ma <mctt90@gmail.com>
parent d6d517bb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/pinsage/README.md
View file @ 3ebf2660
...@@ -25,7 +25,7 @@ interacted. The distance between two items are measured by Euclidean distance o ...@@ -25,7 +25,7 @@ interacted. The distance between two items are measured by Euclidean distance o
item embeddings, which are learned as outputs of PinSAGE. item embeddings, which are learned as outputs of PinSAGE.
``` ```
python model.py data.pkl --num-epochs 300 --num-workers 2 --device cuda:0 data.pkl --hidden-dims 64 python model.py data.pkl --num-epochs 300 --num-workers 2 --device cuda:0 --hidden-dims 64
``` ```
The HITS@10 is 0.01241, compared to 0.01220 with SLIM with the same dimensionality. The HITS@10 is 0.01241, compared to 0.01220 with SLIM with the same dimensionality.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment