imyutaro/fm

Factorization Machine

PyTorch implementation of Factorization Machine(FM) and BASKET-SENSITIVE FACTORIZATION MACHINE(BFM).

• Paper
  • Factorization Machines
  • Basket-Sensitive Personalized Item Recommendation

How to build and run a docker container.

CPU ver with docker-compose

# Building a container
./start.sh
# Enter a docker container
sudo docker exec -it fm zsh

GPU ver

# Building a container
./start_gpu.sh
# Enter a docker container
sudo docker exec -it fm_gpu zsh