Open HotBento opened 10 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保存。
屏幕截图
实验环境(请补全下列信息):
@HotBento 你好!感谢你提出的建议,我们会在下一个开发计划中完善KG-based dataloader。
描述这个 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保存。
屏幕截图
实验环境(请补全下列信息):