jyonn / gnrs

news representation learning

GreenRec: Green AI Benchmarking for News Recommendation

⭐ GreenRec is implemented under the Legommenders framework. All the experiments can be reproduced with Legommenders. This repository will no longer be maintained, thanks!

Environment

pip install -r requirements.txt

Data Processing

Please specify the path to the data in python file

cd process/mind
python processor.py

Configuration

Data

Please refer to config_v2/data/mind.yaml for the data configuration.

Model

We support the following models on both MIND small and large datasets:

	NAML	LSTUR	NRMS	DCN	DIN	BST
ID-based	ID-NAML	ID-LSTUR	ID-NRMS	DCN	DIN	BST
text-based	NAML	LSTUR	NRMS	text-DCN	text-DIN	text-BST
PLMNR	PLMNR-NAML	PLMNR-LSTUR	PLMNR-NRMS	PLMNR-DCN	PLMNR-DIN	PLMNR-BST
BERT	BERT-NAML	BERT-LSTUR	BERT-NRMS	BERT-DCN	BERT-DIN	BERT-BST
MFT	MFT-NAML	MFT-LSTUR	MFT-NRMS	MFT-DCN	MFT-DIN	MFT-BST

Training and Testing

python worker.py 
    --config config/data/mind.yaml 
    --model config/model/nrms.yaml 
    --exp config/exp/tt-nrms.yaml
    --embed config/embed/null.yaml
    --version small-v2