createDummyFeatures gives wrong and unconsistant names used with two factor levels

[Sade154]

createDummyFeatures gives wrong and unconsistant names used with two factor levels

[pat-s]

Thanks for reporting. Looks like a bug on the first look.
Not sure when I'll have some time to look at it.

You might want to try dummy encoding in {mlr3pipelines} and use {mlr3} in general, I'd think/hope that it works just as expected there.

[Sade154]

Thanks, I will try mlr3pipelines then.
Regarding mlr, a small change in the function createDummyFeatures.data.frame fixed the issue for me.

# initial code in mlr
# if (method == "reference" && length(work.cols) == length(dummies)) {
#   colnames(dummies) = Map(function(col, pre) {
#     stri_paste(pre, tail(levels(col), -1), sep = ".")
#   }, obj[work.cols], prefix)
# }

# changed version
if (method == "reference") {
    colnames(dummies) = unlist(Map(function(col, pre) {
      stri_paste(pre, tail(levels(col), -1), sep = ".")
    }, obj[work.cols], prefix))
}