Skip to content

My implementation of Factorization Machine in PyTorch.

Notifications You must be signed in to change notification settings

gennadylaptev/FM_in_PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

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.

About

My implementation of Factorization Machine in PyTorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published