R/GroupProfiles.R

In trajeR: Group Based Modeling Trajectory

#' Profiles of each group
#'
#' \code{GroupProfiles} calculate the profile of a group regarding covariate. It is a
#' cross tabulation of individual level trajectory group assignments with
#' individual level characteristic that might be associated with trajectory
#' group membership.
#'
#' @param sol Trajectory's object. A object of type trajectory.
#' @param Y Matrix. A matrix containing the variables in the model.
#' @param A Matrix. A matrix containing the time variable data.
#' @param X Matrix. An optional matrix that modify the probability of belong to group.
#'   By default its value is a matrix with one column  with value 1.
#'
#' @return A table of real.
#' @export
#'
#' @examples
#' data <- read.csv(system.file("extdata", "CNORM2gr.csv", package = "trajeR"))
#' data <- as.matrix(data)
#' sol <- trajeR(
#'   Y = data[, 2:6], A = data[, 7:11], Risk = data[, 12],
#'   degre = c(2, 2), Model = "CNORM", Method = "L"
#' )
#' GroupProfiles(sol, Y = data[, 2:6], A = data[, 7:11], X = data[, 12])
GroupProfiles <- function(sol, Y, A, X) {
  prob <- GroupProb(sol, Y = Y, A = A, X = X)
  gr <- sapply(1:nrow(prob), function(s) {
    which.max(prob[s, ])
  })
  if (!is.matrix((X))) {
    X <- matrix(X)
  }
  tab <- c()
  for (j in 1:ncol(X)) {
    tabl <- c()
    for (i in 1:sol$groups) {
      tabl <- c(tabl, mean(X[gr == i, j]))
    }
    tab <- rbind(tab, tabl)
  }
  colnames(tab) <- paste0("Gr ", 1:sol$groups)
  rownames(tab) <- colnames(X)
  return(tab)
}