search

summmeer / session-based-news-recommendation

source code of paper "Positive, Negative and Neutral: Modeling Implicit Feedback in Session-based News Recommendation", which is accepted at SIGIR 2022.
31 stars 8 forks source link

Missing files on MIND dataset #8

Open LUUUUUUZ opened 1 year ago

LUUUUUUZ commented 1 year ago

Could you share some missing files on MIND dataset? python mind_preprocess.py --content_info It seems ../data/mind/articles_embeddings_1.pkl is missing, which I thought maybe is generated from get_content_vec.py. But how could I load the pre-trained model?

Besides, python data_process/generate_neighbor.py It seems publish_time_1.txt is missing?

I will really appreciate it if you could provide above two files. Thank you!

summmeer commented 1 year ago

articles_embeddings_1.pkl: articles_embeddings_1.zip

publish_time_1.txt can be generated in mind_preprocess.py.

LUUUUUUZ commented 1 year ago

Thanks! Have a good day ;)

LUUUUUUZ commented 1 year ago

Sorry to bother again. It seems MIND/articles_category.pkl is also missing. Although it's simple to generate but I don't know the form of this data structure. I will appreciate it if you could provide this file. Thanks!

LUUUUUUZ commented 1 year ago

And could you share your environment? like python version and tf version?

summmeer commented 1 year ago

articles_category is a dictionary, pseudo code:

articles_category = {}
for each news_id, news_category:
    articles_category[news_id] = news_category
pickle.dump(articles_category, open('../articles_category.pkl', 'wb'))

Environment info: Python: 3.6.9, TensorFlow: 1.14.0

LUUUUUUZ commented 1 year ago

Thank you so much! It's so nice of you to help me with so many problems. Enjoy your weekend! :)