R/util_plot.R

In pierrejacob/bayeshscore: Bayesian Model Comparison using the Hyvarinen Score

#------------------------------------------------------------------------------#
#---------------------------  Some useful for plotting graphs -----------------#
#------------------------------------------------------------------------------#
#'@rdname simulateData
#'@title simulateData
#'@description This function generates artificial data from a model.
#'@export
simulateData = function(model,theta,nobservations) {
  # If state-space model, generate latent states then observations
  if (!is.null(model$dimX)){
    X <- matrix(nrow = model$dimX, ncol = nobservations)
    Y <- matrix(nrow = model$dimY, ncol = nobservations)
    if (nobservations >= 1){
      X[,1] <- model$rinitial(theta,1)
      Y[,1] <- model$robs(X[,1,drop=FALSE],1,theta)
    }
    if (nobservations >= 2){
      for (t in 2:nobservations) {
        X[,t] = model$rtransition(X[,t-1,drop=FALSE], t, theta)
        Y[,t] = model$robs(X[,t,drop=FALSE], t, theta)
      }
    }
    return (list(X = X, Y = Y))
  }
  # If the model makes it possible, generate observations directly
  else {
    Y = matrix(nrow = model$dimY, ncol = nobservations)
    if (nobservations >= 1) {
      Y[,1] = model$rpredictive(1,1,theta,NULL)
    }
    if (nobservations >= 2) {
      for (t in 2:nobservations) {
        Y[,t] = model$rpredictive(1,t,theta,Y[,1:(t-1),drop=FALSE])
      }
    }
    return (Y)
  }
}
#------------------------------------------------------------------------------#
#'@rdname setmytheme
#'@title Customize graphical settings
#'@description This function customizes the theme used by ggplot.
#'@export
setmytheme <- function(){
  theme_set(theme_bw())
  theme_update(axis.text.x = element_text(size = 20),
               axis.text.y = element_text(size = 20),
               axis.title.x = element_text(size = 25, margin=margin(20,0,0,0)),
               axis.title.y = element_text(size = 25, angle = 90, margin = margin(0,20,0,0)),
               legend.text = element_text(size = 20),
               legend.title = element_text(size = 20),
               title = element_text(size = 30),
               strip.text = element_text(size = 25),
               strip.background = element_rect(fill="white"),
               panel.margin = unit(2, "lines"),
               legend.position = "bottom")
}