/FM_in_PyTorch

My implementation of Factorization Machine in PyTorch.

Primary LanguageJupyter Notebook

Factorization Machine in PyTorch

My implementation of Factorization Machine in PyTorch.

The present notebook has also been published as a Kaggle kernel.

Data

The data I used in my experiments can be found in this dataset on Kaggle. The original source of the data is mlcourse.ai: Dota 2 Winner Prediction. I just extracted hero ids for each of the teams and one-hot encoded them.