MANNeR

A NEW VERSION OF THE FRAMEWORK IS OUT IN NewsRecLib

Description

This is the implementation of the MANNeR framework from the paper "Train Once, Use Flexibly: A Modular Framework for Multi-Aspect Neural News Recommendation"

How to run

Install dependencies

# clone project
git clone https://github.com/andreeaiana/manner
cd manner

# [OPTIONAL] create conda environment
conda create -n manner_env python=3.9
conda activate manner_env

# install pytorch according to instructions
# https://pytorch.org/get-started/

# install requirements
pip install -r requirements.txt

Train model with chosen experiment configuration from configs/experiment/

python manner/train.py experiment=experiment_name.yaml

You can override any parameter from command line like this

python manner/train.py trainer.max_epochs=20 data.batch_size=64

# train on GPU
python manner/train.py trainer=gpu

Run ensemble (CR-Module + A-Module) with trained sub-modules with chosen experiment configuration from configs/experiment/

python manner/train.py experiment=experiment_name.yaml train=False

Citation

@misc{iana2023train,
      title={Train Once, Use Flexibly: A Modular Framework for Multi-Aspect Neural News Recommendation}, 
      author={Andreea Iana and Goran Glavaš and Heiko Paulheim},
      year={2023},
      eprint={2307.16089},
      archivePrefix={arXiv},
      primaryClass={cs.IR}
}