Skip to content
forked from JinyuZ1996/RL-ISN

Implementation of "Reinforcement Learning-enhanced Shared-account Cross-domain Sequential Recommendation" (TKDE 2022)

Notifications You must be signed in to change notification settings

otakbeku/RL-ISN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation

RL-ISN

Overall description

  • Here presents the code of RL-ISN. As the dataset is too big for GitHub, we upload two datasets (i.e., Hvideo and Hamazon) on Bitbucket: https://bitbucket.org/jinyuz1996/rl-isn-data/src/main/. You should download them before training the RL-ISN. The code is attached to our paper: " Reinforcement Learning-enhanced Shared-account Cross-domain Sequential Recommendation" (TKDE 2022) ". If you want to use our codes and datasets in your research, please cite our paper. Note that, our paper is still in the state of 'Early Access', if you want to see the preview version (on IEEE Xplore), please visit: https://ieeexplore.ieee.org/document/9802739.

Code description

Vesion of implements and tools

  1. python 3.6
  2. tensorflow 1.12.0
  3. scipy 1.5.3
  4. numpy 1.16.0
  5. pandas 0.22.0
  6. matplotlib 3.3.4
  7. Keras 1.0.7
  8. tqdm 4.60.0

Source code of RL-ISN

  1. the definition of Basic recommender see: RL_ISN/ISN_recommender.py
  2. the definition of Agent see: RL_ISN/ISN_agent.py
  3. the definition of Enviroment see: RL_ISN/ISN_environment.py
  4. the definition of Training process see: RL_ISN/ISN_train.py
  5. the definition of Evaluating process see: RL_ISN/ISN_evaluation.py
  6. the preprocess of dataset see: RL_ISN/ISN_configuration.py
  7. the hyper-parameters of RL-ISN see: RL_ISN/ISN_parameters.py
  8. to run the training method see: RL_ISN/ISN_main.py and the training log printer was defined in: RL_ISN/ISN_printer.py
    • The directory named Checkpoint is used to save the trained recommenders ,agents and joint-training models.
    • Note that, for different dataset (i.e., HVIDEO and HAMAZON), we set two different settings for the haper-parameters as 'ISN_parameters_Hamazon.py' and 'ISN_parameters_Hvideo.py'

About

Implementation of "Reinforcement Learning-enhanced Shared-account Cross-domain Sequential Recommendation" (TKDE 2022)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%