NewsRecLib：一个基于 PyTorch-Lightning 的神经新闻推荐库
信息检索
2023-10-03 v1
摘要
NewsRecLib 是一个基于 Pytorch-Lightning 和 Hydra 开发的开源库，用于训练和评估神经新闻推荐模型。NewsRecLib 的首要目标是通过（i）为详尽实验研究提供统一且高度可配置的框架，以及（ii）实现对不同模型架构组件和训练机制的性能贡献的透彻分析，来促进可复现研究和严谨的实验评估。NewsRecLib 高度模块化，允许在单一配置文件中指定实验，并包含广泛的日志功能。此外，NewsRecLib 提供了若干知名神经模型、训练方法、标准评估基准以及新闻推荐评估指标的即用型实现。
引用
@article{arxiv.2310.01146,
title = {NewsRecLib: A PyTorch-Lightning Library for Neural News Recommendation},
author = {Andreea Iana and Goran Glavaš and Heiko Paulheim},
journal= {arXiv preprint arXiv:2310.01146},
year = {2023}
}
备注
Accepted at the 2023 Conference on Empirical Methods in Natural Language Processing (EMNLP 2023)