Skip to content

nickwzk/InvRL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Invariant Representation Learning for Multimedia Recommendation

This our official implementation via Pytorch for the paper:

Xiaoyu Du, Zike Wu, Fuli Feng, Xiangnan He and Jinhui Tang. Invariant Representation Learning for Multimedia Recommendation. In ACM MM`22, October 10–14, 2022, Lisboa, Portugal.

Requirements

  • Python==3.8.10
  • Pytorch==1.11.0+cu113
  • numpy, scipy, argparse, logging, sklearn, tqdm

Example to Run the Codes

First, we run UltraGCN for pretraining:

python main.py --model UltraGCN

After that, we run the InvRL model:

python main.py --model InvRL

About

The official implementation of InvRL

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages