X-Git-Url: https://git.auder.net/?a=blobdiff_plain;f=R%2FgenerateIO.R;h=83d8cc9f19bcc4f3689817547cd5e9db7b83c23b;hb=d1531659214edd6eaef0ac9ec835455614bba16c;hp=9e84af5a282d4f81d683c38946cd5b7a15ebb364;hpb=493a35bfea6d1210c94ced8fbfe3e572f0389ea5;p=valse.git

diff --git a/R/generateIO.R b/R/generateIO.R
index 9e84af5..83d8cc9 100644
--- a/R/generateIO.R
+++ b/R/generateIO.R
@@ -1,6 +1,14 @@
-library(MASS) #simulate from a multivariate normal distribution
-
-generateIO = function(meanX, covX, covY, pi, beta, n){ #don't need meanX
+#' Generate a sample of (X,Y) of size n
+#' @param covX covariance for covariates
+#' @param covY covariance for the response vector
+#' @param pi   proportion for each cluster
+#' @param beta regression matrix
+#' @param n    sample size
+#' @return list with X and Y
+#' @export
+#-----------------------------------------------------------------------
+generateIO = function(covX, covY, pi, beta, n)
+{
   size_covX = dim(covX)
   p = size_covX[1]
   k = size_covX[3]
@@ -11,15 +19,17 @@ generateIO = function(meanX, covX, covY, pi, beta, n){ #don't need meanX
   Y = matrix(0,n,m)
   BX = array(0, dim=c(n,m,k))
   
-  for(i in 1:n){
-    for(r in 1:k){
+  require(MASS) #simulate from a multivariate normal distribution
+  for (i in 1:n)
+  {
+    for (r in 1:k)
+    {
       BXir = rep(0,m)
-      for(mm in 1:m){
+      for (mm in 1:m)
         Bxir[[mm]] = X[i,] %*% beta[,mm,r]
-      }
-      Y[i,]=Y[i,] + pi[[r]] * mvrnorm(1,BXir, covY[,,r])
+      Y[i,] = Y[i,] + pi[r] * mvrnorm(1,BXir, covY[,,r])
     }
   }
   
-  return(list(X,Y))
-}
\ No newline at end of file
+  return (list(X=X,Y=Y))
+}