Unofficial PyTorch implementation of Attentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networks (github repo), as well as FM, on the MovieLens dataset.
- models should allow arbitrary number of dense features
- TensorBoard Support