pytorch-fm

Factorization Machine models in PyTorch.

This package provides an implementation of various factorization machine models and common datasets in PyTorch.

Minimal requirements

• Python 3.x

• PyTorch 1.1.0

• numpy

• lmdb

Installation

Install with pip:

pip install torchfm

API documentation

• torchfm package
  • torchfm.dataset
    • torchfm.dataset.avazu
    • torchfm.dataset.criteo
    • torchfm.dataset.movielens
  • torchfm.model
    • torchfm.model.afi
    • torchfm.model.afm
    • torchfm.model.dcn
    • torchfm.model.dfm
    • torchfm.model.ffm
    • torchfm.model.fm
    • torchfm.model.fnfm
    • torchfm.model.fnn
    • torchfm.model.lr
    • torchfm.model.nfm
    • torchfm.model.pnn
    • torchfm.model.wd
    • torchfm.model.xdfm
  • torchfm.layer

Indices and tables

• Index

• Module Index

• Search Page