Skip to content

just4jin/fm

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

Factorization Machines (FM)

Movie ratings prediction model trained by FM using pyFM library.

Factorization Machines (FM) uses stochastic gradient descent with adaptive regularization as a learning method, which adapts the regularization automatically while training the model parameters. It combines the generality of feature engineering with the superiority of factorization models in estimating interactions between categorical variables of large domain. FM is especially appealing for prediction problems over the data sets with large categorical variables because it allows to estimate variable interactions reliably due to factorized interactions.

References

[1] Steffen Rendle (2012): Factorization Machines with libFM, in ACM Trans. Intell. Syst. Technol., 3(3), May.

[2] Steffen Rendle: Learning recommender systems with adaptive regularization. WSDM 2012: 133-142

About

trying out factorization machines (fm) algorithm

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages