Skip to content

Latest commit

 

History

History
16 lines (13 loc) · 525 Bytes

README.md

File metadata and controls

16 lines (13 loc) · 525 Bytes

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