search

BUPT-GAMMA / OpenHGNN

This is an open-source toolkit for Heterogeneous Graph Neural Network(OpenHGNN) based on DGL.
Apache License 2.0
848 stars 142 forks source link

SLiCE model #118

Closed Shreyas-Bhat closed 7 months ago

Shreyas-Bhat commented 2 years ago

❓ Questions and Help

Hello! Thank you for the wonderful library.

I needed some help in running the code for the SLiCE model on any dataset for link prediction that you might have tested on. Can you please help me with the steps of doing the same?

VoidHaruhi commented 2 years ago

Hi @Shreyas-Bhat , I'm the contributor of SLiCE model. Since this model is too complex, I failed to reproduce the performance in the original paper. I recommend you refer to the repository of the author. I'm testing how to run SLiCE on OpenHGNN, but I cannot ensure the performance. By the way, I cannot reproduce the results even using the original code. You may ask the author. I guess she use the embedding from CompGCN as the initial embedding, then do pretraining and finetuning.

VoidHaruhi commented 2 years ago

@Shreyas-Bhat We currently only support "amazon4SLICE" dataset for SLiCE model (the dataset is the same as Amazon dataset used for link prediction in the original paper). To run SLiCE, use "python main.py -m SLiCE -d amazon4SLICE -t link_prediction". The embedding file will be generated at "output/SLiCE/amazon/amazon.emb". You can use various embedding methods to check whether the performance can be reproduced, but we only support node2vec pretraining (if you put a file with the correct form of embedding in that directory, the model will not do this pretraining procedure).

Shreyas-Bhat commented 2 years ago

Hi @VoidHaruhi Thank you for the reply! I am facing this error- Can you help me out here?

image
lazishu2000 commented 7 months ago

This issue has been resolved; for specific details, see: https://github.com/BUPT-GAMMA/OpenHGNN/commit/01275b74590425f6afa6ea36004ee18df6f7b7de