search

BUPT-GAMMA / OpenHGNN

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

MetaHIN #240

Open wubin010 opened 1 month ago

wubin010 commented 1 month ago

Description

MetaHIN模型，用元学习解决推荐系统的冷启动，由于原作者并没用用到图结构，我这里就没转dgl图 coldstart_recommendation.py 新加的trainflow coldstart_recommendation.py 新加的task MetaHIN.py 新加的model MetaHIN_dataset.py 新加的dataset config.ini中的input_dir信息我没修改，还用的自己的绝对路径