R/genpar_adtneh2.R

In curesurv: Mixture and Non Mixture Parametric Cure Models to Estimate Cure Indicators

genpar_adtneh2 <- function(theta, z_tau, z_alpha) {
n_z_tau <- ncol(z_tau)
n_z_alpha <- ncol(z_alpha)
n_z_tau_ad <- n_z_tau - 1
n_z_alpha_ad <- n_z_alpha - 1
alpha0 <- theta[1]

if (n_z_tau > 0 & n_z_alpha > 0) {
  alpha_k <- theta[2:(n_z_alpha + 1)]
  beta <- (theta[n_z_alpha + 2])
  tau0 <- theta[n_z_alpha + 2 + 1]
  tau_z <- theta[(n_z_alpha + 2 + 1 + 1):(n_z_alpha + 2 + n_z_tau + 1)]
  return(c(alpha0 = as.numeric(alpha0),
           alpha_k = as.numeric(alpha_k),
           beta = as.numeric(beta),
           tau0 = as.numeric(tau0),
           tau_z = as.numeric(tau_z) ))

}else if (n_z_tau > 0 & n_z_alpha == 0) {
  beta <- (theta[2])
  tau0 <- theta[2 + 1]
  tau_z <- theta[(2 + 1 + 1):(2 + n_z_tau + 1)]

  return(c(alpha0 = as.numeric(alpha0),
           beta = as.numeric(beta),
           tau0 = as.numeric(tau0),
           tau_z = as.numeric(tau_z) ))
}else if (n_z_tau == 0 & n_z_alpha > 0) {
  alpha_k <- theta[2:(n_z_alpha + 1)]
  beta <- (theta[n_z_alpha + 2])
  tau0 <- theta[n_z_alpha + 2 + 1]
  return(c(alpha0 = as.numeric(alpha0),
           alpha_k = as.numeric(alpha_k),
           beta = as.numeric(beta),
           tau0 = as.numeric(tau0)
           ))
}else if (n_z_tau == 0 & n_z_alpha == 0) {
  tau0 <- theta[n_z_alpha + 2 + 1]
  beta <- (theta[n_z_alpha + 2])
  return(c(alpha0 = as.numeric(alpha0),
           beta = as.numeric(beta),
           tau0 = as.numeric(tau0)))
}
}