https://github.com/dmlc/difacto

Distributed Factorization Machines
https://github.com/dmlc/difacto

Last synced: 3 months ago
JSON representation

Distributed Factorization Machines

• Host: GitHub
• URL: https://github.com/dmlc/difacto
• Owner: dmlc
• License: other
• Created: 2015-11-27T21:14:53.000Z (over 9 years ago)
• Default Branch: master
• Last Pushed: 2016-03-23T20:00:14.000Z (over 9 years ago)
• Last Synced: 2025-04-09T20:05:27.511Z (3 months ago)
• Language: C++
• Size: 647 KB
• Stars: 297
• Watchers: 37
• Forks: 92
• Open Issues: 14
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
#  *Distributed Factorization Machines*

[![Build Status](https://travis-ci.org/dmlc/difacto.svg?branch=master)](https://travis-ci.org/dmlc/difacto)

[![codecov.io](https://codecov.io/github/dmlc/difacto/coverage.svg?branch=master)](https://codecov.io/github/dmlc/difacto?branch=master)

[![Documentation Status](https://readthedocs.org/projects/difacto/badge/?version=latest)](http://difacto.readthedocs.org/en/latest/?badge=latest)

[![GitHub license](http://dmlc.github.io/img/apache2.svg)](./LICENSE)

Fast and memory efficient library for factorization machines (FM).

- Supports both ℓ1 regularized logistic regression and factorization

  machines.

- Runs on local machine and distributed clusters.

- Scales to datasets with billions examples and features.

### Quick Start

The following commands clone and build difacto, then download a sample dataset,

and train FM with 2-dimension on it.

```bash

git clone --recursive https://github.com/dmlc/difacto

cd difacto; git submodule update --init; make -j8

./tools/download.sh gisette

build/difacto data_in=data/gisette_scale val_data=data/gisette_scale.t lr=.02 V_dim=2 V_lr=.001

```

### History

Origins from

[wormhole/learn/difacto](https://github.com/dmlc/wormhole/tree/master/learn/difacto).

(NOTE: this project is still under developing)

### References

Mu Li, Ziqi Liu, Alex Smola, and Yu-Xiang Wang.

DiFacto — Distributed Factorization Machines. In WSDM, 2016