by scikit-learn-contrib

scikit-learn-contrib / polylearn

A library for factorization machines and polynomial networks for classification and regression in Py...

212 Stars 40 Forks Last release: Not found BSD 2-Clause "Simplified" License 56 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

.. -- mode: rst --


A library for factorization machines and polynomial networks for classification and regression in Python.

Github repository. 

.. image:: https://travis-ci.org/scikit-learn-contrib/polylearn.svg?branch=master :target: https://travis-ci.org/scikit-learn-contrib/polylearn

.. image:: https://ci.appveyor.com/api/projects/status/g9xnar9081l3vsw7/branch/master?svg=true :target: https://ci.appveyor.com/project/vene/polylearn

.. image:: https://coveralls.io/repos/scikit-learn-contrib/polylearn/badge.svg?branch=master&service=github :target: https://coveralls.io/r/scikit-learn-contrib/polylearn

.. image:: https://circleci.com/gh/scikit-learn-contrib/polylearn/tree/master.svg?style=shield&circle-token=:circle-token :target: https://circleci.com/gh/scikit-learn-contrib/polylearn/

Factorization machines and polynomial networks are machine learning models that can capture feature interaction (co-occurrence) through polynomial terms. Because feature interactions can be very sparse, it's common to use low rank, factorized representations; this way, we can learn weights even for feature co-occurrences that haven't been observed at training time.

Factorization machines are popular for recommender systems, as they are a generalization of matrix completion models.

This package provides:

  • coordinate descent algorithm for fitting factorization machines of degree 2 or 3,
  • coordinate descent algorithm for fitting polynomial networks of arbitrary degree,
  • scikit-learn 
    _-compatible API,
  • Cython 
    _ implementations for computationally intensive parts.


Binary packages are not yet available.

The development version of polylearn can be installed from its git repository. In this case it is assumed that you have a working C++ compiler.

  1. Obtain the sources by::

    git clone https://github.com/scikit-learn-contrib/polylearn.git

or, if

is unavailable,
download as a ZIP from GitHub 
  1. Install the dependencies::

    via pip

    pip install numpy scipy scikit-learn nose pip install sklearn-contrib-lightning

    via conda

    conda install numpy scipy scikit-learn nose conda install -c conda-forge sklearn-contrib-lightning

  2. Build and install polylearn::

    cd polylearn python setup.py build sudo python setup.py install


The solvers implemented are introduced in [1]. Factorization machines are introduced in [2] and polynomial networks in [3]_.

.. [1] Mathieu Blondel, Masakazu Ishihata, Akinori Fujino, Naonori Ueda. Polynomial Networks and Factorization Machines: New Insights and Efficient Training Algorithms. In: Proc. of ICML 2016. [


.. [2] Steffen Rendle. Factorization machines. In: Proc. of IEEE ICDM 2010. [


.. [3] Roi Livni, Shai Shalev-Shwartz, Ohad Shamir. On the computational efficiency of training neural networks. In: Proc. of NIPS 2014. [



  • Vlad Niculae, 2016-present

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.