MKM-SR

Incorporating User Micro-behaviors and Item Knowledge into Multi-task Learning for Session-based Recommendation

Paper data and code

This is the code for the SIGIR2020 Paper:Incorporating User Micro-behaviors and Item Knowledge into Multi-task Learning for Session-based Recommendation. We have implemented our methods in pytorch.

Here are two datasets we used in our paper. After download the datasets, you can put them in the folder: data/

• KKbox: https://www.kaggle.com/c/kkbox-music-recommendation-challenge/data
• JData: https://jdata.jd.com/html/detail.html?id=8 There is a small dataset demo included in the folder data/,which can be used to test the correctness of the code.

We have also inclueded some baseline codes in this paper.

Usage

You need to run the file data/data_prepare.py first to preprocess the data.

For example: cd data; python data_prepare.py --dataset=demo

usage: prepare.py [--dataset Demo][--remove_new_item]
optional arguments:
--dataset DATASET_PATH dataset name: Demo/Jdata/KKbox

Then you can run the file main.py to train the model.

usage: main.py 
optional arguments:
  --dataset             dataset name
  --batchSize           input batch size
  --hiddenSize          hidden state size
  --epoch EPOCH         the number of epochs to train for
  --lr LR               learning rate
  --l2 L2               l2 penalty
  --step STEP           gnn propogation steps
  --patience PATIENCE   the number of epoch to wait before early stop
  --remove_new_items    whether keep new item
  --mode                model mode,there we only keep MKM_SR,if you need other mode, you can contact with us, or use the ablation version.