Learn R Programming

NPflow (version 0.10.1)

plot_DPM: Plot of a Dirichlet process mixture of gaussian distribution partition

Description

Plot of a Dirichlet process mixture of gaussian distribution partition

Usage

plot_DPM(z, U_mu = NULL, U_Sigma = NULL, m, c, i, alpha = "?", U_SS = NULL, dims2plot = 1:nrow(z), ellipses = ifelse(length(dims2plot) < 3, TRUE, FALSE), gg.add = list(theme()))

Arguments

z
data matrix d x n with d dimensions in rows and n observations in columns.
U_mu
either a list or a matrix containing the current estimates of mean vectors of length d for each cluster. Default is NULL in which case U_SS has to be provided.
U_Sigma
either a list or an array containing the d x d current estimates for covariance matrix of each cluster. Default is NULL in which case U_SS has to be provided.
m
vector of length n containing the number of observations currently assigned to each clusters.
c
allocation vector of length n indicating which observation belongs to which clusters.
i
current MCMC iteration number.
alpha
current value of the DP concentration parameter.
U_SS
a list containing "mu" and "S". Default is NULL in which case U_mu and U_Sigma have to be provided.
dims2plot
index vector, subset of 1:d indicating which dimensions should be drawn. Default is all of them.
ellipses
a logical flag indicating whethe ellipses should be drawn around clusters. Default is TRUE if only 2 dimensions are plotted, FALSE otherwise.
gg.add
a list of instructions to add to the ggplot2 instruction. See +.gg. Default is list(theme()), which adds nothing to the plot.