inst/presentations/statisfaction.R

In pierrejacob/montecarlodsm: Gibbs sampler for Dempster-Shafer inference in Categorical distributions

## This script generates two plots
rm(list = ls())
library(dempsterpolytope)
library(latex2exp)
graphsettings <- set_custom_theme()
set.seed(1)

# number of categories
K <- 3
categories <- 1:K
# data 
counts <- c(2,3,1)

## triangle with equal sides
attach(graphsettings)
v1 <- v_cartesian[[1]]; v2 <- v_cartesian[[2]]; v3 <- v_cartesian[[3]]
pt_bar <- c(0.3, 0.25, 0.45)
pt_xy  <- barycentric2cartesian(pt_bar, v_cartesian)
# Now with ggplot2
g <- create_plot_triangle(graphsettings)
g <- add_plot_subsimplex(graphsettings, g, pt_bar, 1, fill = graphsettings$cols[1], alpha = 0.8)
g <- add_plot_subsimplex(graphsettings, g, pt_bar, 2, fill = graphsettings$cols[2], alpha = 0.8)
g <- add_plot_subsimplex(graphsettings, g, pt_bar, 3, fill = graphsettings$cols[3], alpha = 0.8)
meanpi1 <- c(pt_xy[1]/3 + v2[1]/3 + v3[1]/3, pt_xy[2]/3 + v2[2]/3 + v3[2]/3)
g <- g + annotate(geom = "text", x = meanpi1[1], y = meanpi1[2], label = expression(Delta[1](theta)), parse = FALSE, size = 7) 
meanpi2 <- c(pt_xy[1]/3 + v1[1]/3 + v3[1]/3, pt_xy[2]/3 + v1[2]/3 + v3[2]/3)
g <- g + annotate(geom = "text", x = meanpi2[1], y = meanpi2[2], label = expression(Delta[2](theta)), parse = FALSE, size = 7) 
meanpi3 <- c(pt_xy[1]/3 + v2[1]/3 + v1[1]/3, pt_xy[2]/3 + v2[2]/3 + v1[2]/3)
g <- g + annotate(geom = "text", x = meanpi3[1], y = meanpi3[2], label = expression(Delta[3](theta)), parse = FALSE, size = 7) 
g <- g + annotate(geom = 'text', size = 7, x = .56, y = .28, label = TeX("$\\theta$", output = "character"), parse = TRUE)
g

# ggsave(filename = "~/Documents/subsimplices.png", plot = g, width = 5, height = 4)
labelsize <- 5
g <- create_plot_triangle(graphsettings)
g <- g + geom_point(aes(x = pt_xy[1], y = pt_xy[2]))
g <- g + annotate(geom = 'text', size = 7, x = .56, y = .28, label = TeX("$\\theta$", output = "character"), parse = TRUE)
gconstraints <- g
gconstraints <- add_ratioconstraint(graphsettings, gconstraints, pt_bar[2]/pt_bar[1], 1, 2, 
                                    graphsettings$contcols[1])
gconstraints <- add_ratioconstraint(graphsettings, gconstraints, pt_bar[3]/pt_bar[1], 1, 3, 
                                    graphsettings$contcols[1])
gconstraints <- gconstraints + geom_label(data = data.frame(x = 0.85, y = 0.28, label = TeX("$w_1 / w_3 \\leq \\theta_1/\\theta_3$", output = "character")), 
                                          aes(x = x, y = y, label = label), parse = TRUE, col = contcols[1], alpha = 0.75, size = labelsize)
gconstraints <- gconstraints + geom_label(data = data.frame(x = 0.2, y = 0.4, label = TeX("$w_1 / w_2 \\leq \\theta_1/\\theta_2$", output = "character")), 
                                          aes(x = x, y = y, label = label), parse = TRUE, col = contcols[1], alpha = 0.75, size = labelsize)

gconstraints

# ggsave(filename = "~/Documents/Constraints.png", plot = gconstraints, width = 5, height = 4)