TensorFlow implementation of arbitrary order (≥2) Factorization Machines for classification and regression tasks.
tffm is a TensorFlow implementation of arbitrary order Factorization Machines, a machine learning model class designed for supervised learning tasks like classification and regression. It efficiently captures feature interactions of any order while maintaining linear inference time with respect to the number of features. The implementation supports both dense and sparse inputs and integrates with TensorFlow's optimization ecosystem.
Machine learning practitioners and researchers working on recommendation systems, click-through rate prediction, or other supervised learning tasks requiring modeling of high-order feature interactions. Particularly useful for those already using TensorFlow who need factorization machine capabilities.
Provides a production-ready, flexible implementation of arbitrary order factorization machines with TensorFlow's optimization capabilities and scikit-learn-like interface. Offers better computational efficiency than naive implementations while supporting both classification and regression tasks with sparse/dense data.
TensorFlow implementation of an arbitrary order Factorization Machine
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Supports factorization machines of order 2 and higher, enabling capture of complex feature interactions, as demonstrated in the usage example with order=6.
Handles both input types, making it versatile for real-world datasets with many categorical features, as highlighted in the key features and usage with input_type parameters.
Integrates with various TensorFlow optimizers like AdamOptimizer, allowing fine-tuning and TensorBoard logging for monitoring, as shown in the dependencies and example code.
Provides a familiar API similar to scikit-learn models, easing adoption for users in that ecosystem, as noted in the usage section with fit and predict methods.
Relies on TensorFlow 1.0+ and tested on version 1.3, which may cause compatibility issues with modern TensorFlow 2.x installations, requiring additional setup or migration work.
The README recommends reading core.py for help, indicating that official documentation is minimal and may be insufficient for advanced usage or troubleshooting.
Focuses solely on basic factorization machines without extensions like field-aware or deep factorization machines, which might be needed for specialized tasks.