fmpytorch

Factorization machine library

PyTorch implementation of a Factorization Machine module

A PyTorch implementation of a Factorization Machine module in cython.

GitHub

173 stars
10 watching
18 forks
Language: Python
last commit: over 6 years ago

Related projects:

Repository Description Stars
locuslab/pytorch_fft Provides an efficient wrapper around CUDA FFTs for PyTorch transformations 314
hengruo/qanet-pytorch An implementation of QANet with PyTorch for text classification tasks 345
hjmshi/pytorch-lbfgs A PyTorch implementation of L-BFGS optimization algorithm for training neural networks 586
coreylynch/pyfm A Python implementation of a factorization machine algorithm for recommendation systems. 922
vadimkantorov/caffemodel2pytorch A tool for converting Caffe models to PyTorch formats 387
yunlongdong/fcn-pytorch A PyTorch implementation of FCN for semantic segmentation with an easy-to-use interface and pre-trained models. 160
kaiyangzhou/dassl.pytorch A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. 1,217
yassouali/pytorch-segmentation Implementation of semantic segmentation models and datasets using PyTorch 1,686
moskomule/eve.pytorch An implementation of an optimization algorithm inspired by a 2016 research paper 33
peterhinch/micropython-fourier Fast Fourier transform implementation in MicroPython for embedded systems 80
jhoon-oh/fedbabu An implementation of federated learning for image classification tasks 51
felixgwu/img_classification_pk_pytorch A PyTorch project for comparing image classification models and facilitating quick experiment setup 365
knighton/torchplus Enhances PyTorch modules by adding sequence step layers with implicit parentheses and supports custom activation functions. 9
fgxaos/pytorch-innvestigate PyTorch implementation of an explainability technique for deep neural networks 9
gunhochoi/kind-pytorch-tutorial A comprehensive tutorial on building and training PyTorch models using Python 392