rishabhmisra / scalable-variational-bayesian-factorization-machine Goto Github PK
View Code? Open in Web Editor NEWScalable Variational Bayesian inference algorithm for FM is developed which converges faster than the existing state-of-the-art MCMC based inference algorithm. Additionally, a stochastic variational Bayesian algorithm for FM is introduced for large scale learning which utilizes SGD.