Linxyhaha/EQUAL

user_embedding

wangyu0627 opened this issue · 2 comments

请问Dataset.py中下面的嵌入如何获取,似乎是运行某个模型预训练得到的,期待你的回答。
best_user_embedding = np.load(dir_str+'/user_embedding.npy', allow_pickle=True)
best_item_embedding = np.load(dir_str+'/item_embedding.npy', allow_pickle=True)

请问Dataset.py中下面的嵌入如何获取,似乎是运行某个模型预训练得到的,期待你的回答。 best_user_embedding = np.load(dir_str+'/user_embedding.npy', allow_pickle=True) best_item_embedding = np.load(dir_str+'/item_embedding.npy', allow_pickle=True)

你好,user_embedding.npy 和 item_embedding.npy是我们通过预训练CLCRec得到的。我们也提供了三个数据集对应的预训练好的embedding文件在"data/{dataset}/"路径下。:)

感谢你的回答,但是我发现只有Amazon数据集存在这样的嵌入,请问你可以提供完整的三个数据集的嵌入吗,我担心自己预训练的结果难以达到论文宣称的结果,期待你的再次解答。