generateSampleIO: Generate sample inputs-outputs

View source: R/sampleIO.R

generateSampleIOR Documentation

Generate sample inputs-outputs

Description

Generate input matrix X of size nxd and binary output of size n, where Y is subdivided into K groups of proportions p. Inside one group, the probability law P(Y=1) is described by the corresponding column parameter in the matrix beta + intercept b.

Usage

generateSampleIO(n, p, beta, b, link)

Arguments

n

Number of individuals

p

Vector of K(-1) populations relative proportions (sum (<)= 1)

beta

Vectors of model parameters for each population, of size dxK

b

Vector of intercept values (use rep(0,K) for no intercept)

link

Link type; "logit" or "probit"

Value

A list with

  • X: the input matrix (size nxd)

  • Y: the output vector (size n)

  • index: the population index (in 1:K) for each row in X

Examples

# K = 3 so we give first two components of p: 0.3 and 0.3 (p[3] = 0.4)
io <- generateSampleIO(1000, c(.3,.3),
  matrix(c(1,3,-1,1,2,1),ncol=3), c(.5,-1,0), "logit")
io$index[1] #number of the group of X[1,] and Y[1] (in 1...K)


morpheus documentation built on Feb. 16, 2023, 10:01 p.m.