LuoUndergradXJTU / TwiBot-22

Offical repository of TwiBot-22 @ NeurIPS 2022, Datasets and Benchmarks Track.
MIT License
153 stars 43 forks source link

Inference on new users using BotRGCN #20

Closed lkevinzc closed 2 years ago

lkevinzc commented 2 years ago

Hi authors, thanks for the nice work!

May I know if I have a trained model using BotRGCN, how I could handle new users in the inference phase?

For example, the new user does not exist in the original dataset, do I need to get all its n-degree neighbors and construct a graph first before I can use the model? Is there any recommended way of doing so?

Thanks!

BunsenFeng commented 2 years ago

@TamSiuhin can you help with this?

TamSiuhin commented 2 years ago

Hi, thanks for your interest in our work! To enable the inductive inference, you do need to induce a subgraph. I recommend you to implement it with torch_geoemetric.NeighborLoader.