R/plot.GPCE.sparse.R

In GPC: Generalized Polynomial Chaos

plot.GPCE.sparse <- function(x, ylim = c(0, 1), ...) {
  if (!is.null(x$Sensitivity)) {
    p <- x$Args$InputDim
    pch = c(21, 24)
    nodeplot(matrix(x$Sensitivity$S[[1]][1,],ncol=p), xlim = c(1, p + 1), ylim = ylim, pch = pch[1])
    nodeplot(matrix(x$Sensitivity$ST[[1]][1,],ncol=p), xlim = c(1, p + 1), ylim = ylim, labels = FALSE,
             pch = pch[2], at = (1:p)+.3, add = TRUE,...)
    legend(x = "topright", legend = c("main effect", "total effect"), pch = pch)
    
    S <- rbind(x$Sensitivity$S[[1]][1,], x$Sensitivity$ST[[1]][1,] - x$Sensitivity$S[[1]][1,])
    colnames(S) <- paste("X", 1:p, sep = "")
    bar.col <- c("white","grey")
    barplot(S, ylim = ylim, col = bar.col)
    legend("topright", c("main effect", "interactions"), fill = bar.col)    
  }
}