search

DeepGraphLearning / graphvite

GraphVite: A General and High-performance Graph Embedding System
https://graphvite.io
Apache License 2.0
1.21k stars 151 forks source link

Can not reproduce baseline results on Wikidata5M dataset #90

Open chocolate9624 opened 2 years ago

chocolate9624 commented 2 years ago

Thanks for your excellent job!

I have trouble in reproduce the baseline results. The command I input is

graphvite baseline rotate wikidata5m --gpu 4 --cpu 3

However, the MRR result is 0.215, which is not comparable to the results at here (0.29). I install graphvite from source.

The log file is attached.

graphvite_baseline_log.txt

chocolate9624 commented 2 years ago

I also find the size of the trained models for Wikidata5M is 8.0G, while the official pre-trained models are all 9.33G (please check here).