data/demoData.R

In pnnl/qFeature: Extract Features from Continuous or Discrete Time Series

# Create some fake data that can be easily used to illustrate the concepts in the
# qFeature package
demoData <- function() {

  if (!requireNamespace("Smisc", quietly = TRUE)) {
    stop("The 'demoData' require the 'Smisc' package to be installed")
  }
    
  # To create the data for a subject and phase
  singleOb <- function(inVec) {

    n <- inVec[["n"]]
      
    # n = number of observations
    scaleFactor <- stats::rpois(1, lambda = 50)
  
    d <- data.frame(subject = inVec[["subject"]],
                    phase = inVec[["phase"]],
                    contA = stats::rnorm(n, mean = scaleFactor * 10, sd = scaleFactor * 2),
                    contB = log(pmax(1,
                            stats::rnorm(n, mean = c(1:(n %/% 2), seq((n %/% 2 + 1), 1, by = -1)) ^ scaleFactor + 10,
                            sd = scaleFactor * 10))),
                    contC = stats::rgamma(n, shape = 2, scale = scaleFactor),
                    contD = (1:n) ^ 2 * stats::rpois(n, lambda = scaleFactor / 10),
                    contE = stats::rbeta(n, 0.5, 0.5) * scaleFactor,
                    discA = sample(rep_len(1:4, n)),
                    discB = as.logical(stats::rbinom(n, 1, stats::runif(1, 0.3, 0.7))),
                    discC = as.factor(sample(rep_len(letters[10:12], n))),
                    discD = sample(rep_len(letters[24:26], n)),
                    stringsAsFactors = FALSE)
   
    return(d)
  
  } # singleOb
  
  
  # Basic Data set structure
  ds <- expand.grid(subject = as.factor(1:7), phase = letters[5:7])
    
  # Set the seed so results are repeatable
  set.seed(893)
  
  # Add the number of observations for each
  ds$n <- pmax(15, stats::rpois(nrow(ds), 30))

  # Generate the actual data
  for (i in 1:nrow(ds)) {

    assign(paste("d", i, sep = ""), singleOb(ds[i,]))
      
  }

  # Row bind them all together at once
  return(Smisc::qbind(paste("d", 1:nrow(ds), sep = "")))

} # demoData

# Create the dataframe
demoData <- demoData()