Skip to content

Latest commit

 

History

History
38 lines (26 loc) · 1.11 KB

README.md

File metadata and controls

38 lines (26 loc) · 1.11 KB

theano-bpr

A library implementing Bayesian Personalised Ranking (BPR) for Matrix Factorisation, as described by Rendle et al. in :

http://arxiv.org/abs/1205.2618

This model tries to predict a personalised ranking of items from a user's viewing history. It has been shown to be very efficient for recommendation tasks. It's also used in a variety of other tasks, such as matrix completion, link prediction and tag recommendation.

This library uses Theano and can therefore run on a GPU through CUDA or on the CPU, for which you'll need a working BLAS. We recommend using OpenBlas.

Installation

$ pip install theano-bpr

Usage

An iPython Notebook demonstrating the use of theano-bpr over the Movielens dataset is available in examples/.

Testing

$ nosetests

Licensing terms and authorship

See 'COPYING' and 'AUTHORS' files