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
..
dataset fix #11, set the param "header=None" when the dataset is MovieLens 1M 4 years ago
model Implemented high order factorization machines 3 years ago
__init__.py add pytorch-fm 4 years ago
layer.py #24 fixed a comment on MultiLayerPerceptron forward function 3 years ago

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

Python other