如何在 R 中对多个分类变量进行一次热编码

我正在研究一个预测问题，并且正在 R 中构建一棵决策树，我有几个分类变量，我想在我的训练和测试集中对它们进行一致的单热编码。 我设法在我的训练数据上做到了这一点：

temps <- X_train
tt <- subset(temps, select = -output)
oh <- data.frame(model.matrix(~ . -1, tt), CLASS = temps$output)

但我找不到在我的测试集上应用相同编码的方法，我该怎么做？

我建议使用 caret 包中的 dummyVars 函数：

customers <- data.frame(
  id=c(10, 20, 30, 40, 50),
  gender=c('male', 'female', 'female', 'male', 'female'),
  mood=c('happy', 'sad', 'happy', 'sad','happy'),
  outcome=c(1, 1, 0, 0, 0))
customers
id gender  mood outcome
1 10   male happy       1
2 20 female   sad       1
3 30 female happy       0
4 40   male   sad       0
5 50 female happy       0


# dummify the data
dmy <- dummyVars(" ~ .", data = customers)
trsf <- data.frame(predict(dmy, newdata = customers))
trsf
id gender.female gender.male mood.happy mood.sad outcome
1 10             0           1          1        0       1
2 20             1           0          0        1       1
3 30             1           0          1        0       0
4 40             0           1          0        1       0
5 50             1           0          1        0       0

example source

您可以对训练集和验证集应用相同的过程。