Auto-Surprise:基于 Parzen 估计树(TPE)优化的自动化推荐系统(AutoRecSys)库
信息检索
2020-09-01 v1 机器学习
机器学习
摘要
我们介绍 Auto-Surprise,一个自动化推荐系统库。Auto-Surprise 是 Surprise 推荐系统库的扩展,简化了算法选择与配置过程。与使用开箱即用的 Surprise 库相比,Auto-Surprise 在 MovieLens、Book Crossing 和 Jester 数据集上评估时表现更优,并可能选出运行时间显著更低的算法。与 Surprise 的网格搜索相比,Auto-Surprise 在 RMSE 上表现相当或略好,且在寻找最优超参数时明显更快。
引用
@article{arxiv.2008.13532,
title = {Auto-Surprise: An Automated Recommender-System (AutoRecSys) Library with Tree of Parzens Estimator (TPE) Optimization},
author = {Rohan Anand and Joeran Beel},
journal= {arXiv preprint arXiv:2008.13532},
year = {2020}
}
备注
To be presented at RecSys '20 Fourteenth ACM Conference on Recommender Systems, September 21-26, 2020, Virtual Event