RUCAIBox / RecBole

A unified, comprehensive and efficient recommendation library
https://recbole.io/
MIT License
3.37k stars 606 forks source link

[🐛BUG] knowledge graph based recommender system在保存dataloader时会报错 #1966

Open HotBento opened 8 months ago

HotBento commented 8 months ago

描述这个 bug knowledge graph based recommender system在保存dataloader时会报错。 将config['save_dataloaders']设置为非None值后,程序会报错'KnowledgeBasedDataLoader' object has no attribute 'generator'。 我看了下源代码,发现recbole.data.utils中的save_split_dataloaders (line 76)函数只支持继承了AbstractDataLoader的dataloader,而KG-based recommender system使用的KnowledgeBasedDataLoader类并没有继承AbstractDataLoader类。希望能够支持KG-based recommender system的dataloader保存。

如何复现 复现这个 bug 的步骤: from recbole.quick_start import run_recbole run_recbole('KGAT', 'ml-100k', config_dict={'save_dataloaders':True})

预期 希望能够支持KG-based recommender system的dataloader保存。

屏幕截图 image

实验环境(请补全下列信息):

BishopLiu commented 8 months ago

@HotBento 你好!感谢你提出的建议,我们会在下一个开发计划中完善KG-based dataloader。