Skip to content

shurain/lightfm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LightFM

An implementation of Factorization Machines. Intended to be educational.

Currently supports naive feature vectorization using maps and feature hashing a.k.a. hashing tricks.

Weights are optimized using stochastic gradient descent. It only looks at a single training example and does not work in full-batch or mini-batch mode.

Data

LightFM expects the following input format.

target feature:weight feature:weight

First term is the target which is followed by features. An example is given below.

3.5 p54081:1 i843:1
3 p49752:1 i324:1
4.5 p60777:1 i1011:1

Build

make

Test

make test

Execution

LightFM takes various command line arguments. You need to specify the path for training data and test data.

bin/lightfm -d <training data path> -t <test data path>

Other arguments can be inspected by --help command.

bin/lightfm --help

Reference

About

Clone of factorization machines

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published