Skip to content

An implementation of Bayesian Personalised Ranking in Theano

License

Notifications You must be signed in to change notification settings

DSE-capstone-sharknado/theano-bpr

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

An implementation of Bayesian Personalised Ranking in Theano

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%