← all repositories

geffy/tffm

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

778 stars Jupyter Notebook ML Frameworks
tffm
Velocity · 7d
+0.2
★ / day
Trend
steady
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.

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