Skip to content

This is our implementation of Explainable Recommendation System for Solving Review Loss

License

Notifications You must be signed in to change notification settings

n60512/HANN-Plus

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Explainable Recommendation System for Solving Review Loss

This is our implementation for the paper:

Explainable Recommendation System for Solving Review Loss

Author: Sean Chen (n60512@gmail.com)

ABSTRACT

We proposed a review-base recommender system named HANN-Plus, a hierarchical attention neural network to model user’s preference and product’s preference. HANN-Plus composed of two sub-models. The first one is rating prediction model named HANN-RPM, we adjust the calculation method of attention mechanism to improve the reviews’ extraction performance of model. The second one is review generation model named HANN-RGM, which is based on encoder-decoder architecture and can be used to generate the representation for making user aware of why such products are recommended.

Training flow

hann-train

Rating Prediction Model

hann-rpm

Review Generation Model

RGM

Environments

  • Python 3
  • Pytoch
  • tqdm
  • gensim
  • numpy
  • pymysql

Dataset

In this experiments, we use the datasets from Amazon prodct data. (http://jmcauley.ucsd.edu/data/amazon/)

About

This is our implementation of Explainable Recommendation System for Solving Review Loss

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published