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

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.