Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change the way functions are called

Browse Source
This commit is contained in:
unknown 2015-11-07 22:04:54 +01:00
parent 77ae180d3d
commit 996645dc17
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
R-package/R/xgb.plot.tree.R
View File

@ -70,7 +70,7 @@ xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NU
allTrees[, shape:= "rectangle"][Feature == "Leaf", shape:= "oval"] allTrees[, shape:= "rectangle"][Feature == "Leaf", shape:= "oval"]
allTrees[, filledcolor:= "Beige"][Feature == "Leaf", filledcolor:= "Khaki"] allTrees[, filledcolor:= "Beige"][Feature == "Leaf", filledcolor:= "Khaki"]
nodes <- create_nodes(nodes = allTrees[,ID], nodes <- DiagrammeR::create_nodes(nodes = allTrees[,ID],
label = allTrees[,label], label = allTrees[,label],
#type = c("lower", "lower", "upper", "upper"), #type = c("lower", "lower", "upper", "upper"),
style = "filled", style = "filled",
@ -81,7 +81,7 @@ xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NU
fontname = "Helvetica" fontname = "Helvetica"
) )
edges <- create_edges(from = allTrees[Feature != "Leaf", c(ID)] %>% rep(2), edges <- DiagrammeR::create_edges(from = allTrees[Feature != "Leaf", c(ID)] %>% rep(2),
to = allTrees[Feature != "Leaf", c(Yes, No)], to = allTrees[Feature != "Leaf", c(Yes, No)],
label = allTrees[Feature != "Leaf", paste("<",Split)] %>% c(rep("",nrow(allTrees[Feature != "Leaf"]))), label = allTrees[Feature != "Leaf", paste("<",Split)] %>% c(rep("",nrow(allTrees[Feature != "Leaf"]))),
color = "DimGray", color = "DimGray",
@ -90,11 +90,11 @@ xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NU
fontname = "Helvetica", fontname = "Helvetica",
rel = "leading_to") rel = "leading_to")
graph <- create_graph(nodes_df = nodes, graph <- DiagrammeR::create_graph(nodes_df = nodes,
edges_df = edges, edges_df = edges,
graph_attrs = "rankdir = LR") graph_attrs = "rankdir = LR")
render_graph(graph, width = width, height = height) DiagrammeR::render_graph(graph, width = width, height = height)
} }
# Avoid error messages during CRAN check. # Avoid error messages during CRAN check.