X-Git-Url: https://git.auder.net/?p=morpheus.git;a=blobdiff_plain;f=pkg%2FR%2FsampleIO.R;h=6fa38aefac234bd01f3c9358d2733bd5ad4b6d70;hp=753e00d3ee1c86ab6f04e609072b1adfe6bd4e36;hb=778a955e6830f48306f2e35322440c5f9a0adaf3;hpb=01bd5c5b56b403b5487e232603643307d3821ed0

diff --git a/pkg/R/sampleIO.R b/pkg/R/sampleIO.R
index 753e00d..6fa38ae 100644
--- a/pkg/R/sampleIO.R
+++ b/pkg/R/sampleIO.R
@@ -51,7 +51,7 @@ generateSampleIO = function(n, p, β, b, link)
 	{
 		index = c(index, rep(i, classes[i]))
 		newXblock = cbind( MASS::mvrnorm(classes[i], zero_mean, id_sigma), 1 )
-		arg_link = newXblock %*% β
+		arg_link = newXblock %*% β[,i] #β
 		probas =
 			if (link == "logit")
 			{
@@ -61,10 +61,10 @@ generateSampleIO = function(n, p, β, b, link)
 			else #"probit"
 				pnorm(arg_link)
 		probas[is.nan(probas)] = 1 #overflow of exp(x)
-		probas = rowSums(p * probas)
+		#probas = rowSums(p * probas)
 		X = rbind(X, newXblock)
-		Y = c( Y, vapply(probas, function(p) (ifelse(p >= .5, 1, 0)), 1) )
-		#Y = c( Y, vapply(probas, function(p) (rbinom(1,1,p)), 1) )
+		#Y = c( Y, vapply(probas, function(p) (ifelse(p >= .5, 1, 0)), 1) )
+		Y = c( Y, vapply(probas, function(p) (rbinom(1,1,p)), 1) )
 	}
 	shuffle = sample(n)
 	# Returned X should not contain an intercept column (it's an argument of estimation