← all repositories

geffy/tffm

A TensorFlow implementation of arbitrary-order Factorization Machines for classification and regression tasks.

777 stars Jupyter Notebook ML Frameworks
tffm
Not currently ranked — collecting fresh signals.
star history

This repository provides a TensorFlow-based implementation of Factorization Machines supporting arbitrary order interactions (>=2). It handles dense and sparse inputs, supports gradient-based optimization, and offers logistic and MSE loss functions for classification and regression. The library follows a scikit-learn-like interface for model training and evaluation.

Frequently asked

What is geffy/tffm?
A TensorFlow implementation of arbitrary-order Factorization Machines for classification and regression tasks.
Is tffm open source?
Yes — geffy/tffm is open source, released under the MIT license.
What language is tffm written in?
geffy/tffm is primarily written in Jupyter Notebook.
How popular is tffm?
geffy/tffm has 777 stars on GitHub.
Where can I find tffm?
geffy/tffm is on GitHub at https://github.com/geffy/tffm.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.