Blei Lab
Blei Lab
edward
A probabilistic programming language in TensorFlow. Deep generative models, variational inference.
causal-text-embeddings
Software and data for "Using Text Embeddings for Causal Inference"
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.
dtm
This implements topics that change over time (Dynamic Topic Models) and a model of how individual documents predict that change.
hdp
Hierarchical Dirichlet processes. Topic models where the data determine the number of topics. This implements Gibbs sampling.
lda-c
This is a C implementation of variational EM for latent Dirichlet allocation (LDA), a topic model for text or other discrete data.
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)
ars-reparameterization
Source code for Naesseth et. al. "Reparameterization Gradients through Acceptance-Rejection Sampling Algorithms" (2017)
class-slda
Implements supervised topic models with a categorical response.