Update vignette + optim code

[morpheus.git] / pkg / R / plot.R

diff --git a/pkg/R/plot.R b/pkg/R/plot.R
index 8acd2c0..29a254e 100644 (file)
--- a/pkg/R/plot.R
+++ b/pkg/R/plot.R
@@ -48,14 +48,14 @@ plotHist <- function(mr, x, y)
 #' @examples
 #' #See example in ?plotHist
 #' @export
-plotBox <- function(mr, x, y)
+plotBox <- function(mr, x, y, xtitle="")
 {
        params <- extractParam(mr, x, y)
        L = length(params)
        # Plot boxplots side by side
        par(mfrow=c(1,L), cex.axis=1.5, cex.lab=1.5, mar=c(4.7,5,1,1))
        for (i in 1:L)
-               boxplot(params[[i]], ylab="Parameter value")
+               boxplot(params[[i]], xlab=xtitle, ylab="Parameter value")
 }
 
 #' plotCoefs
@@ -69,7 +69,7 @@ plotBox <- function(mr, x, y)
 #' @examples
 #' #See example in ?plotHist
 #' @export
-plotCoefs <- function(mr, params, idx)
+plotCoefs <- function(mr, params, idx, xtitle="Parameter")
 {
        L <- nrow(mr[[1]][[1]])
        K <- ncol(mr[[1]][[1]])
@@ -94,7 +94,7 @@ plotCoefs <- function(mr, params, idx)
        avg_param <- as.double(params_hat)
        std_param <- as.double(stdev)
        matplot(cbind(params[o],avg_param[o],avg_param[o]+std_param[o],avg_param[o]-std_param[o]),
-               col=1, lty=c(1,5,3,3), type="l", lwd=2, xlab="Parameter", ylab="Value")
+               col=1, lty=c(1,5,3,3), type="l", lwd=2, xlab=xtitle, ylab="")
 
        #print(o) #not returning o to avoid weird Jupyter issue... (TODO:)
 }