Skip to content

ACL'21 Oral, Personalized Transformer for Explainable Recommendation

Notifications You must be signed in to change notification settings

lileipisces/PETER

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PETER (PErsonalized Transformer for Explainable Recommendation)

Paper

A large pretrained GPT-2 version is available at PEPLER!

A small ecosystem for Recommender Systems-based Natural Language Generation is available at NLG4RS!

Datasets to download

  • TripAdvisor Hong Kong
  • Amazon Movies & TV
  • Yelp 2019

For those who are interested in how to obtain (feature, opinion, template, sentiment) quadruples, please refer to Sentires-Guide.

Usage

Below are examples of how to run PETER (with and without the key feature).

python -u main.py \
--data_path ../TripAdvisor/reviews.pickle \
--index_dir ../TripAdvisor/1/ \
--cuda \
--checkpoint ./tripadvisorf/ \
--peter_mask \
--use_feature >> tripadvisorf.log

python -u main.py \
--data_path ../TripAdvisor/reviews.pickle \
--index_dir ../TripAdvisor/1/ \
--cuda \
--checkpoint ./tripadvisor/ \
--peter_mask >> tripadvisor.log

Code dependencies

  • Python 3.6
  • PyTorch 1.6

Code references

Citation

@inproceedings{ACL21-PETER,
	title={Personalized Transformer for Explainable Recommendation},
	author={Li, Lei and Zhang, Yongfeng and Chen, Li},
	booktitle={ACL},
	year={2021}
}

About

ACL'21 Oral, Personalized Transformer for Explainable Recommendation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages