RUCAIBox / RecBole

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

请问如何对测试集进行划分 #2005

Open yin214 opened 7 months ago

yin214 commented 7 months ago

例如我想测试训练好的序列推荐模型在男user和女user上的性能差异,该怎么实现?

TayTroye commented 7 months ago

@yin214 你好 你可以通过load_data_and_model加载训练好的模型和测试集,再自行对测试集进行划分。

yin214 commented 7 months ago

@TayTroye 感谢,利用load_data_and_model,我已经成功将l测试集进行划分。 我想请问load_data_and_model函数返回的三个dataloader和当时训练时的dataloader是完全一致的嘛?