Skip to content
This repository was archived by the owner on May 24, 2018. It is now read-only.

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

Distributed Factorization Machine

For a p dimension example x, FM models the data by

hat_y

where w_i denotes by the i-th element of the p-length vector w, and v_i denotes by the i-th row of the p-by-k matrix V.

Given training data pairs (x,y), FM learns the model w and V by solving the following objective:

obj

Here is the loss function such as logistic loss.

Quick start

Go the root directory of wormhole, then build by make difacto. Next try a small dataset using 2 worker and 1 server:

tracker/dmlc_local.py -n 2 -s 1 bin/difacto.dmlc learn/difacto/guide/demo.conf

More