ptree_criteria: ptree_criteria

In jwolfson/visparty: Visualize conditional regression trees

Usage

ptree_criteria(newtree, node_id, left)

Arguments

newtree
node_id
left

Examples

##---- Should be DIRECTLY executable !! ----
##-- ==>  Define data, use random,
##--	or do  help(data=index)  for the standard data sets.

## The function is currently defined as
function (newtree, node_id, left) 
{
    if (nodes(newtree, node_id)[[1]]$terminal) {
        return("(error: terminal node)")
    }
    if (nodes(newtree, node_id)[[1]]$psplit$ordered) {
        sp <- nodes(newtree, node_id)[[1]]$psplit$splitpoint
        vn <- nodes(newtree, node_id)[[1]]$psplit$variableName
        if (left) {
            op <- "<="
        }
        else {
            op <- ">"
        }
        return(paste(vn, op, sp))
    }
    else {
        psplit <- nodes(newtree, node_id)[[1]]$psplit
        if (left) {
            l <- as.logical(psplit$splitpoint)
        }
        else {
            l <- as.logical(!psplit$splitpoint)
        }
        r <- paste(attr(psplit$splitpoint, "levels")[l], sep = "", 
            collapse = "','")
        return(paste(psplit$variableName, " in ('", r, "')", 
            sep = ""))
    }
  }

jwolfson/visparty documentation built on May 20, 2019, 6:27 a.m.