An open API service indexing awesome lists of open source software.

https://github.com/bbc/theano-bpr

An implementation of Bayesian Personalised Ranking in Theano
https://github.com/bbc/theano-bpr

Last synced: 9 months ago
JSON representation

An implementation of Bayesian Personalised Ranking in Theano

Awesome Lists containing this project

README

          

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](http://deeplearning.net/software/theano/) and
can therefore run on a [GPU through CUDA](http://deeplearning.net/software/theano/tutorial/using_gpu.html) or on the CPU,
for which you'll need a working BLAS. We recommend using [OpenBlas](http://www.openblas.net).

Installation
------------

$ pip install theano-bpr

Usage
-----

An iPython Notebook demonstrating the use of theano-bpr over the
Movielens dataset [is available in examples/](http://nbviewer.ipython.org/github/bbcrd/theano-bpr/blob/master/examples/example.ipynb).

Testing
-------

$ nosetests

Licensing terms and authorship
------------------------------

See 'COPYING' and 'AUTHORS' files