X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=pkg%2FR%2Fplot.R;h=9727493e183cbdd4680d00c4a24c799a7120ce79;hb=6dd5c2acccd10635449230faa824b7e8906911bf;hp=0097607bf99d2449239e70fec13164fb9201714c;hpb=5fc1b9d9bbb20ebf5228792f5885b77991c0cec9;p=morpheus.git

diff --git a/pkg/R/plot.R b/pkg/R/plot.R
index 0097607..9727493 100644
--- a/pkg/R/plot.R
+++ b/pkg/R/plot.R
@@ -6,10 +6,10 @@
 #
 extractParam <- function(mr, x=1, y=1)
 {
-	# Obtain L vectors where L = number of res lists in mr
-	lapply( mr, function(mr_list) {
-		sapply(mr_list, function(m) m[x,y])
-	} )
+  # Obtain L vectors where L = number of res lists in mr
+  lapply( mr, function(mr_list) {
+    sapply(mr_list, function(m) m[x,y])
+  } )
 }
 
 #' plotHist
@@ -21,7 +21,7 @@ extractParam <- function(mr, x=1, y=1)
 #' @param y Column index of the element inside the aggregated parameter
 #'
 #' @examples
-#' \dontrun{
+#' \donttest{
 #' β <- matrix(c(1,-2,3,1),ncol=2)
 #' mr <- multiRun(...) #see bootstrap example in ?multiRun : return lists of mu_hat
 #' μ <- normalize(β)
@@ -31,12 +31,12 @@ extractParam <- function(mr, x=1, y=1)
 #' @export
 plotHist <- function(mr, x, y)
 {
-	params <- extractParam(mr, x, y)
-	L = length(params)
-	# Plot histograms 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)
-		hist(params[[i]], breaks=40, freq=FALSE, xlab="Parameter value", ylab="Density")
+  params <- extractParam(mr, x, y)
+  L = length(params)
+  # Plot histograms 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)
+    hist(params[[i]], breaks=40, freq=FALSE, xlab="Parameter value", ylab="Density")
 }
 
 #' plotBox
@@ -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")
+  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]], xlab=xtitle, ylab="Parameter value")
 }
 
 #' plotCoefs
@@ -64,50 +64,39 @@ plotBox <- function(mr, x, y)
 #'
 #' @inheritParams plotHist
 #' @param params True value of parameters matrix
+#' @param idx List index to process in mr
 #'
 #' @examples
 #' #See example in ?plotHist
 #' @export
-plotCoefs <- function(mr, params)
+plotCoefs <- function(mr, params, idx, xtitle="Parameter")
 {
-	nf <- length(mr)
-	L <- nrow(mr[[1]][[1]])
-	K <- ncol(mr[[1]][[1]])
+  L <- nrow(mr[[1]][[1]])
+  K <- ncol(mr[[1]][[1]])
 
-	params_hat <- vector("list", nf)
-	stdev <- vector("list", nf)
-	for (i in 1:nf)
-	{
-		params_hat[[i]] <- matrix(nrow=L, ncol=K)
-		stdev[[i]] <- matrix(nrow=L, ncol=K)
-	}
-	for (x in 1:L)
-	{
-		for (y in 1:K)
-		{
-			estims <- extractParam(mr, x, y)
-			for (i in 1:nf)
-			{
-				params_hat[[i]][x,y] <- mean(estims[[i]])
-#				stdev[[i]][x,y] <- sqrt( mean( (estims[[i]] - params[x,y])^2 ) )
-				# HACK remove extreme quantile in estims[[i]] before computing sd()
-				stdev[[i]][x,y] <- sd( estims[[i]] [ estims[[i]] < max(estims[[i]]) & estims[[i]] > min(estims[[i]]) ] )
-			}
-		}
-	}
+  params_hat <- matrix(nrow=L, ncol=K)
+  stdev <- matrix(nrow=L, ncol=K)
+  for (x in 1:L)
+  {
+    for (y in 1:K)
+    {
+      estims <- extractParam(mr, x, y)
+      params_hat[x,y] <- mean(estims[[idx]])
+#      stdev[x,y] <- sqrt( mean( (estims[[idx]] - params[x,y])^2 ) )
+      # HACK remove extreme quantile in estims[[i]] before computing sd()
+      stdev[x,y] <- sd( estims[[idx]] ) #[ estims[[idx]] < max(estims[[idx]]) & estims[[idx]] > min(estims[[idx]]) ] )
+    }
+  }
 
-	par(mfrow=c(1,nf), cex.axis=1.5, cex.lab=1.5, mar=c(4.7,5,1,1))
-	params <- as.double(params)
-	o <- order(params)
-	for (i in 1:nf)
-	{
-		avg_param <- as.double(params_hat[[i]])
-		std_param <- as.double(stdev[[i]])
-		matplot(cbind(params[o],avg_param[o],avg_param[o]+std_param[o],avg_param[o]-std_param[o]),
-			col=c(2,1,1,1), lty=c(1,1,2,2), type="l", lwd=2, xlab="param", ylab="value")
-	}
+  par(cex.axis=1.5, cex.lab=1.5, mar=c(4.7,5,1,1))
+  params <- as.double(params)
+  o <- order(params)
+  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=xtitle, ylab="")
 
-	#print(o) #not returning o to avoid weird Jupyter issue... (TODO:)
+  #print(o) #not returning o to avoid weird Jupyter issue... (TODO:)
 }
 
 #' plotQn
@@ -115,25 +104,28 @@ plotCoefs <- function(mr, params)
 #' Draw 3D map of objective function values
 #'
 #' @param N Number of starting points
+#' @param n Number of points in sample
+#' @param p Vector of proportions
+#' @param b Vector of biases
 #' @param β Regression matrix (target)
 #' @param link Link function (logit or probit)
 #'
 #' @export
 plotQn <- function(N, n, p, β, b, link)
 {
-	d <- nrow(β)
-	K <- ncol(β)
-	io <- generateSampleIO(n, p, β, b, link)
-	op <- optimParams(K, link, list(X=io$X, Y=io$Y))
-	# N random starting points gaussian (TODO: around true β?)
-	res <- matrix(nrow=d*K+1, ncol=N)
-	for (i in seq_len(N))
-	{
-		β_init <- rnorm(d*K)
-		par <- op$run( c(rep(1/K,K-1), β_init, rep(0,K)) )
-		par <- op$linArgs(par)
-		Qn <- op$f(par)
-		res[,i] = c(Qn, par[K:(K+d*K-1)])
-	}
-	res #TODO: plot this, not just return it...
+  d <- nrow(β)
+  K <- ncol(β)
+  io <- generateSampleIO(n, p, β, b, link)
+  op <- optimParams(K, link, list(X=io$X, Y=io$Y))
+  # N random starting points gaussian (TODO: around true β?)
+  res <- matrix(nrow=d*K+1, ncol=N)
+  for (i in seq_len(N))
+  {
+    β_init <- rnorm(d*K)
+    par <- op$run( c(rep(1/K,K-1), β_init, rep(0,K)) )
+    par <- op$linArgs(par)
+    Qn <- op$f(par)
+    res[,i] = c(Qn, par[K:(K+d*K-1)])
+  }
+  res #TODO: plot this, not just return it...
 }