Merge pull request #50 from henrifnk/master

change as_precrec prediction
mlr-org · Sep 16, 2020 · a5a5437 · a5a5437
2 parents 8d1f98c + 951357d
commit a5a5437
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/R/BenchmarkResult.R b/R/BenchmarkResult.R
@@ -33,8 +33,8 @@
 #'
 #' head(fortify(object))
 #' autoplot(object)
-#' autoplot(object$clone()$filter(task_ids = "spam"), type = "roc")
-#' autoplot(object$clone()$filter(task_ids = "pima"), type = "prc")
+#' autoplot(object$clone(deep = TRUE)$filter(task_ids = "spam"), type = "roc")
+#' autoplot(object$clone(deep = TRUE)$filter(task_ids = "pima"), type = "prc")
 autoplot.BenchmarkResult = function(object, # nolint
   type = "boxplot",
   measure = NULL,

diff --git a/R/as_precrec.R b/R/as_precrec.R
@@ -66,7 +66,7 @@ as_precrec.BenchmarkResult = function(object) { # nolint
     stopf("Unable to convert benchmark results with multiple resamplings.")
   }
 
-  predictions = map(scores$prediction, "test")
+  predictions = scores$prediction
   data = transpose_list(map(predictions, roc_data))
   data$labels = split(data$labels, scores$iteration)
   data$scores = split(data$scores, scores$iteration)

diff --git a/man/autoplot.BenchmarkResult.Rd b/man/autoplot.BenchmarkResult.Rd
diff --git a/man/autoplot.LearnerClassifRpart.Rd b/man/autoplot.LearnerClassifRpart.Rd