alumbreras / MMLE-GaP

Code for our ICML '18 paper "Closed-form Marginal Likelihood in Gamma-Poisson Matrix Factorization"

Home Page:http://proceedings.mlr.press/v80/filstroff18a.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Maximum Marginal Likelihood for Dictionary Learning in Gamma-Poisson models (MMLE-GaP)

Code related to the paper:

Closed-form Marginal Likelihood in Gamma-Poisson Matrix Factorization
Filstroff L., Lumbreras A., FĂ©votte C.
International Conference on Machine Learning (2018)

R/Rcpp implementation of MMLE-GaP and related algorithms.

The main MCMC-EM algorithm is in R/MMLE_MCEM.R. From this file, calls are made to methods in other files that compute the Monte Carlo E-Step and the M-step. The basic MC E-Step is a Gibbs sampler, but we have also played with other alternatives. There are R and Rcpp implementations (src/ folder) that are much faster.

About

Code for our ICML '18 paper "Closed-form Marginal Likelihood in Gamma-Poisson Matrix Factorization"

http://proceedings.mlr.press/v80/filstroff18a.html


Languages

Language:R 65.0%Language:C++ 34.1%Language:C 0.6%Language:Makefile 0.3%