ivoryRabbit / mvae-pytorch

Multi Variational AutoEncoder with Pytorch

MVAE-Pytorch For Marketing Campaign

Paper

• Variational Autoencoders for Collaborative Filtering
• On Mitigating Popularity Bias in Recommendations via Variational Autoencoders

Dataset

• A Purchase Dataset in S3

How To Run

1. Install

pip3 install -r requirements.txt

2. Load Data

python3 load_data.py

3. Preprocess

python3 preprocess.py

4. Train

python3 train.py

5. Inference

python3 inference.py