三、SR-GNN模型定义 classGNN(Module): def__init__(self,hidden_size,step=1):#输入仅需确定隐状态数和步数 super(GNN,self).__init__() self.step=step#gnn前向传播的步数default=1 self.hidden_size=hidden_size self.input_size=hidden_size*2 self.gate_size=3*hidden_size #有关Parameter函数的...
Session-based Recommendation with Graph Neural Networks 官方代码代码仓库: https://github.com/CRIPAC-DIG/SR-GNN#paper-data-and-code 有torch和tf两个版本数据集下载一个是YOOCHOOSE,仓库的链接失效了,去…
Session-based Recommendation with Graph Neural Networks,AAAI-2019,论文代码的中文注释版本,官方代码地址:https://github.com/CRIPAC-DIG/SR-GNN