bethunebtj / glint-fm

Factorization Machines on Spark and Glint

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Glint FM

Factorization Machines on Spark and Glint

An implementation of distributed factorization machines on Spark using the Glint parameter server.

To build, run sbt package. You will need to have Glint and spark-libFM installed locally, since they are not available on Maven central.

You will need to clone the glint repository and run sbt publish-local (master branch should be fine).

For spark-libFM you will need to build it with Spark 2.0 support - check out my branch here which was used for performance comparisons. Again, run sbt publish-local to install locally before running glint-fm.

About

Factorization Machines on Spark and Glint

License:Apache License 2.0


Languages

Language:Scala 100.0%