converts factors with levels T,F to logical, RWeka does this

mlr-org · Aug 26, 2015 · 8e4ed73 · 8e4ed73
1 parent ea387ec
commit 8e4ed73
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/R/readARFF.R b/R/readARFF.R
@@ -89,7 +89,12 @@ readARFF = function(path, data.reader = "readr", tmp.file = tempfile(), show.inf
     ct = header$col.types[i]
     if (ct == "factor") {
       clevs = header$col.levels[[i]]
-      dat[,i] = factor(dat[,i], levels = clevs)
+      # RWEKA parses this to logical
+      # FIXME: DOC THIS!
+      if (identical(clevs, c("TRUE", "FALSE")) || identical(clevs, c("FALSE", "TRUE")))
+        dat[,i] = as.logical(dat[,i])
+      else
+        dat[,i] = factor(dat[,i], levels = clevs)
     }
   }
   })