# 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