图神经网络在推荐系统中得到了广泛的应用。从早期的GCMC和NGCF,到后期在阿里/微信/京东/快手/抖音等工业界落地的大规模实践。尽管用起来很好用,但是涉及到代码实现,数据存储和线上服务,还是有一些困难。然后,还没有一个框架能够整合和复现常见GNN推荐算法。这个时候,GRecX及时的出现了 。
https://github.com/maenzhier/GRecX
基于 TensorFlow GNN 库 tf_geometric,GRecX 精心实现了多种基于 GNN 的推荐模型。一键run,轻松复现论文结果。
下面是GRecX的结构设计图~主要是基于TensorFlow和 tf_geometric做支撑,然后设计了数据管理,向量搜索,测评&指标等等~