search

THUwangcy / ReChorus

“Chorus” of recommendation models: a light and flexible PyTorch framework for Top-K recommendation.
MIT License
545 stars 91 forks source link

CFKG模型 #14

Closed hzau96yhz closed 3 years ago

hzau96yhz commented 3 years ago

想问一下CFKG这个模型，构建的KG数据集是在哪个文件里，也是在data目录里吗

hzau96yhz commented 3 years ago

文档里说得item_meta.csv（可选） 其格式： itemid \t i \t ... \t r_ \t ... 可选，仅某些型号（CFKG，SLRC +，Chorus，KDA）需要。

但里边是这样的 item_id i_category r_complement r_substitute 您能说明一下后边这两个r_complement r_substitute表示什么含义吗？

THUwangcy commented 3 years ago

表示商品之间的关系，和当前 item_id 有关系的商品以列表形式存在对应列中。比如 item_id 为 1，r_complememt 为 [2,3]，表示对于互补这个关系来说，1 的互补品有 2 和 3，(1, complement, 2) 和 (1, complement, 3) 两个三元组在 KG 中。因为这个里面都是商品之间的关系，就用这种方式存储了。构建 KG 的过程可以参考 KGReader 的实现。