R/utils.R

In msats5/capstoneProject:

#' @export

GetNodeLabel <- function(node) {
  if(node$isLeaf) {
    label = node$class
  } else {
    label = node$feature
  }
}

#' @export

GetEdgeLabel <- function(node) {
  if(!node$isRoot) {
    label = node$name
  }
}