1 Star 0 Fork 0

PeijieSun / U-ARM

加入 Gitee
与超过 800 万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
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

简介

Peijie Sun, Le Wu, Kun Zhang, Yu Su, Meng Wang. An Unsupervised Aspect-aware Recommendation Model with Explanation Text Generation, TOIS. 对应的code. 展开 收起
Python
MulanPSL-2.0
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
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

搜索帮助