search

Linxyhaha / EQUAL

Implementation of paper: Equivariant Learning for Out-of-Distribution Cold-start Recommendation. (backbone model CLCRec) (MM'23)
2 stars 0 forks source link

user_embedding #3

Open wangyu0627 opened 4 months ago

wangyu0627 commented 4 months ago

请问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)

Linxyhaha commented 4 months ago

请问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}/"路径下。:)

wangyu0627 commented 4 months ago

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