File: BenchmarkResultOrderLevels.R

package info (click to toggle)
r-cran-mlr 2.19.2%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 8,264 kB
  • sloc: ansic: 65; sh: 13; makefile: 5
file content (29 lines) | stat: -rw-r--r-- 989 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
# order levels of task.ids in a BenchmarkResult or a similar data.frame
# useful for plotting in ggplot2
# if order.tsks is NULL, just return the df
orderBMRTasks = function(bmr, df = NULL, order.tsks) {
  if (is.null(df)) {
    df = as.data.frame(bmr)
  }
  if (!is.null(order.tsks)) {
    assertCharacter(order.tsks, len = length(getBMRTaskIds(bmr)))
    assertSetEqual(order.tsks, getBMRTaskIds(bmr), ordered = FALSE)
    df$task.id = factor(df$task.id, order.tsks)
  }
  return(df)
}

# order levels of learner.ids of a BenchmarkResult or similar data.frame
# useful for plotting in ggplot2
# if order.tsks is NULL, just return the df
orderBMRLrns = function(bmr, df = NULL, order.lrns) {
  if (is.null(df)) {
    df = as.data.frame(bmr)
  }
  if (!is.null(order.lrns)) {
    assertCharacter(order.lrns, len = length(getBMRLearnerIds(bmr)))
    assertSetEqual(order.lrns, getBMRLearnerIds(bmr), ordered = FALSE)
    df$learner.id = factor(df$learner.id, order.lrns)
  }
  return(df)
}