blei-lab Goto Github PK

ars-reparameterization

Source code for Naesseth et. al. "Reparameterization Gradients through Acceptance-Rejection Sampling Algorithms" (2017)

causal-text-embeddings

Software and data for "Using Text Embeddings for Causal Inference"

class-slda

Implements supervised topic models with a categorical response.

collabtm

collaborative topic modeling

context-selection-embedding

Context Selection for Embedding Models

ctm-c

This implements variational inference for the correlated topic model.

ctr

Collaborative modeling for recommendation. Implements variational inference for a collaborative topic models. These models recommend items to users based on item content and other users' ratings.

deconfounder_public

deconfounder_tutorial

deep-exponential-families

Deep exponential families (DEFs)

diln

This implements the discrete infinite logistic normal, a Bayesian nonparametric topic model that finds correlated topics.

dtm

This implements topics that change over time (Dynamic Topic Models) and a model of how individual documents predict that change.

dynamicpoissonfactorization

Dynamic version of Poisson Factorization (dPF). dPF captures the changing interest of users and the evolution of items over time according to user-item ratings.

edward

A probabilistic programming language in TensorFlow. Deep generative models, variational inference.

embvis

Tool for visualizing and browsing over data point embeddings

expo-mf

Exposure Matrix Factorization: modeling user exposure in recommendation

factorial-network-models

Discussion of Durante et al for JSM 2017. Includes factorial network model generalization.

hdp

Hierarchical Dirichlet processes. Topic models where the data determine the number of topics. This implements Gibbs sampling.

hlda

This implements hierarchical latent Dirichlet allocation, a topic model that finds a hierarchy of topics. The structure of the hierarchy is determined by the data.

lda-bump-cpp

Latent Dirichlet allocation (LDA) with bumping variational inference.

lda-c

This is a C implementation of variational EM for latent Dirichlet allocation (LDA), a topic model for text or other discrete data.

markovian-score-climbing

ml-field-experiments

online-hdp

Online inference for the Hierarchical Dirichlet Process. Fits hierarchical Dirichlet process topic models to massive data. The algorithm determines the number of topics.

onlineldavb

Online variational Bayes for latent Dirichlet allocation (LDA)

poisson-influence-factorization

publications

The pdf and LaTeX for each paper (and sometimes the code and data used to generate the figures).

r-lda-deprecated

The old version of the latent Dirichlet allocation package for R

riken_tutorial

synthetic_controls