All Projects → scikit-learn-contrib → Polylearn

scikit-learn-contrib / Polylearn

Licence: bsd-2-clause
A library for factorization machines and polynomial networks for classification and regression in Python.

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Polylearn

Daisyrec
A developing recommender system in pytorch. Algorithm: KNN, LFM, SLIM, NeuMF, FM, DeepFM, VAE and so on, which aims to fair comparison for recommender system benchmarks
Stars: ✭ 280 (+26.13%)
Mutual labels:  factorization-machines
Tffm
TensorFlow implementation of an arbitrary order Factorization Machine
Stars: ✭ 761 (+242.79%)
Mutual labels:  factorization-machines
Fwumious wabbit
Fwumious Wabbit, fast on-line machine learning toolkit written in Rust
Stars: ✭ 96 (-56.76%)
Mutual labels:  factorization-machines
Attentional factorization machine
TensorFlow Implementation of Attentional Factorization Machine
Stars: ✭ 362 (+63.06%)
Mutual labels:  factorization-machines
Deepctr
Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
Stars: ✭ 5,686 (+2461.26%)
Mutual labels:  factorization-machines
Attentional Neural Factorization Machine
Attention,Factorization Machine, Deep Learning, Recommender System
Stars: ✭ 39 (-82.43%)
Mutual labels:  factorization-machines
Tensorflow Xnn
Tensorflow implementation of DeepFM variant that won 4th Place in Mercari Price Suggestion Challenge on Kaggle.
Stars: ✭ 263 (+18.47%)
Mutual labels:  factorization-machines
Rsparse
Fast and accurate machine learning on sparse matrices - matrix factorizations, regression, classification, top-N recommendations.
Stars: ✭ 145 (-34.68%)
Mutual labels:  factorization-machines
Lightctr
Lightweight and Scalable framework that combines mainstream algorithms of Click-Through-Rate prediction based computational DAG, philosophy of Parameter Server and Ring-AllReduce collective communication.
Stars: ✭ 644 (+190.09%)
Mutual labels:  factorization-machines
Rankfm
Factorization Machines for Recommendation and Ranking Problems with Implicit Feedback Data
Stars: ✭ 71 (-68.02%)
Mutual labels:  factorization-machines
Openlearning4deeprecsys
Some deep learning based recsys for open learning.
Stars: ✭ 383 (+72.52%)
Mutual labels:  factorization-machines
Pytorch Fm
Factorization Machine models in PyTorch
Stars: ✭ 455 (+104.95%)
Mutual labels:  factorization-machines
Deepmatch
A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.
Stars: ✭ 1,051 (+373.42%)
Mutual labels:  factorization-machines
Ytk Learn
Ytk-learn is a distributed machine learning library which implements most of popular machine learning algorithms(GBDT, GBRT, Mixture Logistic Regression, Gradient Boosting Soft Tree, Factorization Machines, Field-aware Factorization Machines, Logistic Regression, Softmax).
Stars: ✭ 337 (+51.8%)
Mutual labels:  factorization-machines
Flurs
🌊 FluRS: A Python library for streaming recommendation algorithms
Stars: ✭ 97 (-56.31%)
Mutual labels:  factorization-machines
Xlearn
High performance, easy-to-use, and scalable machine learning (ML) package, including linear model (LR), factorization machines (FM), and field-aware factorization machines (FFM) for Python and CLI interface.
Stars: ✭ 2,968 (+1236.94%)
Mutual labels:  factorization-machines
Fastfm
fastFM: A Library for Factorization Machines
Stars: ✭ 908 (+309.01%)
Mutual labels:  factorization-machines
Deeptables
DeepTables: Deep-learning Toolkit for Tabular data
Stars: ✭ 207 (-6.76%)
Mutual labels:  factorization-machines
Fmg
KDD17_FMG
Stars: ✭ 116 (-47.75%)
Mutual labels:  factorization-machines
Ctr model zoo
some ctr model, implemented by PyTorch, such as Factorization Machines, Field-aware Factorization Machines, DeepFM, xDeepFM, Deep Interest Network
Stars: ✭ 55 (-75.23%)
Mutual labels:  factorization-machines

.. -- mode: rst --

polylearn

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

Github repository. <https://github.com/scikit-learn-contrib/polylearn/>_

.. 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 <http://scikit-learn.org>_-compatible API,
  • Cython <http://cython.org>_ implementations for computationally intensive parts.

Installation

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 git is unavailable, download as a ZIP from GitHub <https://github.com/scikit-learn-contrib/polylearn/archive/master.zip>_.

  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

References

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. [PDF <http://mblondel.org/publications/mblondel-icml2016.pdf>_]

.. [2] Steffen Rendle. Factorization machines. In: Proc. of IEEE ICDM 2010. [PDF <https://www.ismll.uni-hildesheim.de/pub/pdfs/Rendle2010FM.pdf>_]

.. [3] Roi Livni, Shai Shalev-Shwartz, Ohad Shamir. On the computational efficiency of training neural networks. In: Proc. of NIPS 2014. [arXiv <http://arxiv.org/abs/1410.1141>_]

Authors

  • Vlad Niculae, 2016-present
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].