hopefully found the error (name of targets was not taken)

mlr-org · Aug 11, 2016 · 365d047 · 365d047
1 parent f460a68
commit 365d047
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/R/MultilabelDBRWrapper.R b/R/MultilabelDBRWrapper.R
@@ -62,6 +62,7 @@ predictLearner.MultilabelDBRWrapper = function(.learner, .model, .newdata, ...)
     function(m) getPredictionProbabilities(predict(m, newdata = .newdata, ...), cl = "TRUE")
   }
   pred.lvl1 = sapply(data.frame(asMatrixCols(lapply(models.lvl1, f))), as.numeric)
+  colnames(pred.lvl1) = .model$task.desc$target
   # Meta level prediction
   models.meta = models[(length(.model$task.desc$target) + 1):(2 * length(.model$task.desc$target))]
   newdata.meta = data.frame(.newdata, pred.lvl1)

diff --git a/tests/testthat/test_learners_all_multilabel.R b/tests/testthat/test_learners_all_multilabel.R
@@ -5,7 +5,7 @@ test_that("learners work: multilabel", {
   # settings to make learners faster and deal with small sample size
 hyperpars = list()
 
-# multiabel
+# multilabel
 lrns = mylist("multilabel", create = TRUE)
 lapply(lrns, testThatLearnerCanTrainPredict, task = multilabel.task, hyperpars = hyperpars)