You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
rixwew f74ad19771 #24 fixed a comment on MultiLayerPerceptron forward function 3 years ago
docs delete modules doc dir 4 years ago
examples Implemented high order factorization machines 3 years ago
test Implemented high order factorization machines 3 years ago
torchfm #24 fixed a comment on MultiLayerPerceptron forward function 3 years ago
LICENSE Create LICENSE 4 years ago
README.md Implemented high order factorization machines 3 years ago
requirements.txt add pytorch-fm 4 years ago
setup.py bump up the minor version 4 years ago

This package provides a PyTorch implementation of factorization machine models and common datasets in CTR prediction.

Python other