R/ptree_right.R

In AshwiniKV/visTree: Visualization of Subgroups for Decision Trees

#' Right Split
#'
#' Identifies a node that corresponds to the right split
#'
#' @param newtree Decision tree generated as a party object
#' @param start_id Character vector
#' @keywords pathway decision tree
#' @export


ptree_right <- function(newtree, start_id) {
  node <- as.list(node_party(newtree))
  if (!is.null(node[[start_id]]$kids)) {
    node[[start_id]]$kids[2]
  }
}