Factorization Machines with Theano

A Factorization Machine (FM) is a predictive model that can be used for regression and classification (Rendle 2010). FMs efficiently incorporate pairwise interactions by using factorized parameters.

PyFactorizationMachines is a Theano-based Python implementation of factorization machines.

Update 4/20/2017: The library is now available on PyPI, the Python Package Index. It can be installed with pip.

$ pip install pyfms

For documentation, see documentation.md.

For example usage, see example.py.

This entry was tagged , , , . Bookmark the permalink.

Leave a Reply

Your email address will not be published. Required fields are marked *