This repository contains a Python reimplementation of "Bayesian Personalized Ranking for Novelty Enhancement" [Wasilewski and Hurley, UMAP'19].

Visualize Psudo Data

MODEL_NAME=novelty_enhanced_bpr python main.py novelty_enhanced_bpr.VisualizePsudoData --local-scheduler

Training

MODEL_NAME=novelty_enhanced_bpr python main.py novelty_enhanced_bpr.TrainModel --local-scheduler

Test

MODEL_NAME=novelty_enhanced_bpr python main.py novelty_enhanced_bpr.TestModel --local-scheduler