1 Star 0 Fork 0

PeijieSun / U-ARM

Create your Gitee Account
Explore and code with more than 8 million developers,Free private repositories !:)
Sign up
Clone or Download
contribute
Sync branch
Cancel
Notice: Creating folder will generate an empty file .keep, because not support in Git
Loading...
README.md

An Unsupervised Aspect-aware Recommendation Model withExplanation Text Generation

This repository is the official implementation of [An Unsupervised Aspect-aware Recommendation Model withExplanation Text Generation]. Besides, we also provide the implementation of other baseline models which are used in our paper.

Requirements

To install requirements:

pip install -r requirements.txt

Datasets

The dataset is not contained in this repository. You can download the datasets here:

Training Procedures

Preparation:

  1. git clone this repository
  2. download the datasets
  3. place the datasets to the data directory

Comparative baseline models:

  1. cd src/amazon_pet_supplies/model_name
  2. python train_mdel.py

U-ARM:

  • Pretraining the ABAE model:

    1. cd src/amazon_pet_supplies/abae
    2. python train_abae.py
  • Pretraining the NCF model:

    1. cd src/amazon_pet_supplies/ncf
    2. python train_ncf.py
  • Pretraining the U-ARM rating prediction module:

    1. cd src/amazon_pet_supplies/u_arm_rating
    2. python train_uarm.py
  • Training the U-ARM explanation text generation module:

    1. cd src/amazon_pet_supplies/u_arm_review
    2. python train_uarm.py

About

Peijie Sun, Le Wu, Kun Zhang, Yu Su, Meng Wang. An Unsupervised Aspect-aware Recommendation Model with Explanation Text Generation, TOIS. 对应的code. expand collapse
Python
MulanPSL-2.0
Cancel

Releases

No release

Contributors

All

Activities

Load More
can not load any more
Python
1
https://gitee.com/peijie_hfut/u-arm.git
git@gitee.com:peijie_hfut/u-arm.git
peijie_hfut
u-arm
U-ARM
master

Search