yueyu1030 / SumGNN

[Bioinformatics 2021] This is the repo for the paper `SumGNN: Multi-typed Drug Interaction Prediction via Efficient Knowledge Graph Summarization'.
https://academic.oup.com/bioinformatics/advance-article/doi/10.1093/bioinformatics/btab207/6189090
81 stars 20 forks source link

缩小数据集 #5

Closed Dunant5 closed 3 years ago

Dunant5 commented 3 years ago

请问有什么方法能缩小dataset和KG的规模吗 在个人电脑上需要花很长的时间提取子图 @yueyu1030 @kexinhuang12345

yueyu1030 commented 3 years ago

Wrt. reducing the size for KG: 1) Currently we use KG nodes from the 2-hop neighborhood of the KG. You can reduce the size of the hop during sampling. 2) You can try to remove some specific types of nodes in the KG. (some of them might be irrelevant)

Dunant5 commented 3 years ago

Thanks.