Skip to content

Factorization Machines for Recommendation and Ranking Problems with Implicit Feedback Data

License

Notifications You must be signed in to change notification settings

etlundquist/rankfm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RankFM

Factorization Machines for Recommendation/Ranking Problems with Implicit Feedback Data

RankFM is a pure-python implementation of the general Factorization Machines model class described in (Rendle 2010) adapted for collaborative filtering recommendation/ranking problems with implicit feedback user-item interaction data. It uses the Bayesian Personalized Ranking (BPR-OPT) optimization criteria described in (Rendle 2009) to train model weights via Stochastic Gradient Descent (SGD). Attempts have been made to maintain a sklearn-like interface to the extent possible, and include useful helper/evaluation functions in addition to the main model class.

This package is currently under active development pre-release, and should not yet be considered stable. Release, build status, and PyPI version status will be added once the package gets to a stable and satisfactory state for an initial release. Stay tuned...


under construction