allow plot_min_depth_distribution to work on ranger forests directly

Yue-Jiang · Yue-Jiang · commit 618c6bdd736a · 2019-09-16T13:25:33.000-07:00
diff --git a/R/min_depth_distribution.R b/R/min_depth_distribution.R
@@ -143,7 +143,7 @@ plot_min_depth_distribution <- function(min_depth_frame, k = 10, min_no_of_trees
                                         mean_sample = "top_trees", mean_scale = FALSE, mean_round = 2,
                                         main = "Distribution of minimal depth and its mean"){
   minimal_depth <- NULL; mean_minimal_depth_label <- NULL; mean_minimal_depth <- NULL
-  if("randomForest" %in% class(min_depth_frame)){
+  if(any(c("randomForest", "ranger") %in% class(min_depth_frame))){
     min_depth_frame <- min_depth_distribution(min_depth_frame)
   }
   min_depth_count_list <- min_depth_count(min_depth_frame)

Original file line number	Diff line number	Diff line change
`@@ -143,7 +143,7 @@ plot_min_depth_distribution <- function(min_depth_frame, k = 10, min_no_of_trees`
`143`	`143`	`mean_sample = "top_trees", mean_scale = FALSE, mean_round = 2,`
`144`	`144`	`main = "Distribution of minimal depth and its mean"){`
`145`	`145`	`minimal_depth <- NULL; mean_minimal_depth_label <- NULL; mean_minimal_depth <- NULL`
`146`		`- if("randomForest" %in% class(min_depth_frame)){`
	`146`	`+ if(any(c("randomForest", "ranger") %in% class(min_depth_frame))){`
`147`	`147`	`min_depth_frame <- min_depth_distribution(min_depth_frame)`
`148`	`148`	`}`
`149`	`149`	`min_depth_count_list <- min_depth_count(min_depth_frame)`