generate_linear_dataset {stagedtrees}

Generate a random binary dataset for classification

Description

Randomly generate a simple classification problem.

Usage

generate_linear_dataset(
  p,
  n,
  eps = 1.2,
  gamma = runif(1, min = -p, max = p),
  alpha = runif(p, min = -p, max = p)
)

Arguments

p

number of variables.

n

number of observations.

eps

noise.

gamma

numeric.

alpha

numeric vector of length n.

Value

A data.frame with n independent random variables and one class variable C computed as sign(sum(x * alpha) + runif(1, -eps, eps) + gamma).

Examples

DD <- generate_linear_dataset(p = 5, n = 1000)