R/05e.Assign.R

In btrm: Bayesian Treed Regression Model for Personalized Prediction and Precision Diagnostics

Assign=function(ET,STN){
  ####
  #1. Initial value
  ####
  p=ET$p
  marker=ET$marker
  node.hat=ET$node.hat
  marker.hat=ET$marker.hat
  dir.marker=ET$dir.marker

  ####
  # 2. Update tree structure
  ####
  #2.1 subject index for node STN
  idx=(node.hat==STN)

  #2.2 select marker
  sel.marker=sample(1:p,1,prob=dir.marker)
  marker[STN]=sel.marker
  marker.hat[idx]=marker[STN]

  ####
  # 3. summary
  ####
  ET$marker=marker
  ET$node.hat=node.hat
  ET$marker.hat=marker.hat
  ET$size.cond=TRUE

  return(ET)
}