apply mlr-style

Author: pat-s

DESCRIPTION

@@ -39,15 +39,15 @@ Suggests:
     care,
     FSelectorRcpp,
     lgr,
-    mlr3measures,
     mlr3learners,
+    mlr3measures,
     praznik,
     rpart,
     testthat
 Encoding: UTF-8
 NeedsCompilation: no
 Roxygen: list(markdown = TRUE, r6 = FALSE)
-RoxygenNote: 7.0.1
+RoxygenNote: 7.0.2
 Collate:
     'Filter.R'
     'mlr_filters.R'

R/Filter.R

@@ -80,7 +80,6 @@ Filter = R6Class("Filter",
 
     initialize = function(id, task_type, task_properties = character(), param_set = ParamSet$new(),
       feature_types = character(), packages = character()) {
-
       self$id = assert_string(id)
       self$task_type = assert_subset(task_type, mlr_reflections$task_types$type, empty.ok = FALSE)
       self$task_properties = assert_subset(task_properties, unlist(mlr_reflections$task_properties, use.names = FALSE))
@@ -106,7 +105,6 @@ Filter = R6Class("Filter",
     },
 
     calculate = function(task, nfeat = NULL) {
-
       task = assert_task(as_task(task), feature_types = self$feature_types, task_properties = self$task_properties)
       fn = task$feature_names
 

R/FilterAUC.R

@@ -54,11 +54,12 @@ FilterAUC = R6Class("FilterAUC", inherit = Filter,
 mlr_filters$add("auc", FilterAUC)
 
 
-auc = function (truth, prob) {
+auc = function(truth, prob) {
   n_pos = sum(truth)
   n_neg = length(truth) - n_pos
-  if (n_pos == 0L || n_neg == 0L)
+  if (n_pos == 0L || n_neg == 0L) {
     return(0.5)
+  }
   r = rank(prob, ties.method = "average")
-  (sum(r[truth]) - n_pos * (n_pos + 1L)/2L)/(n_pos * n_neg)
+  (sum(r[truth]) - n_pos * (n_pos + 1L) / 2L) / (n_pos * n_neg)
 }

R/FilterInformationGain.R

@@ -38,7 +38,6 @@
 #' filterGR$param_set$values = list("type" = "gainratio")
 #' filterGR$calculate(task)
 #' head(as.data.table(filterGR), 3)
-#'
 FilterInformationGain = R6Class("FilterInformationGain", inherit = Filter,
   public = list(
     initialize = function() {

man/FilterInformationGain.Rd

@@ -15,8 +15,10 @@ test_that("all regr filters return correct filter values", {
 test_that("Errors for unsupported features", {
   # list filters that only support "numeric" features
   filters = mlr_filters$mget(mlr_filters$keys())
-  filters = Filter(function(x) all(grepl(paste(c("numeric", "integer"),
-      collapse = "|"), x$feature_types)), filters)
+  filters = Filter(function(x) {
+    all(grepl(paste(c("numeric", "integer"),
+      collapse = "|"), x$feature_types))
+  }, filters)
   filters = filters[lengths(filters) != 0]
 
   # supported: numeric, integer