r - 如何将预测值转换为二进制变量并将它们保存到 CSV

Question

我在测试数据上建立了一个决策树模型，然后用它来预测测试数据集中的值。

dtpredict<-predict(ct1, testdat, type="class")

输出如下:

      1       2       3       4       5       6 
    Class_2 Class_2 Class_6 Class_2 Class_8 Class_2 

我想写一个 csv 看起来像:

id, Class_1, Class_2, Class_3, Class_4, Class_5, Class_6, Class_7, Class_8, Class_9
1, 0, 1, 0, 0, 0, 0, 0, 0, 0
2, 0, 1, 0, 0, 0, 0, 0, 0, 0
3, 0, 0, 0, 0, 0, 1, 0, 0, 0
4, 0, 1, 0, 0, 0, 0, 0, 0, 0
5, 0, 0, 0, 0, 0, 0, 0, 1, 0
6, 0, 1, 0, 0, 0, 0, 0, 0, 0

Answer 1

有一个名为 dummies 的包可以很好地完成......

install.packages("dummies")
library(dummies)

x <- factor(c("Class_2", "Class_2", "Class_6", "Class_2", "Class_8", "Class_2"),
            levels = paste("Class", 1:9, sep="_"))

dummy(x, drop = FALSE)

     xClass_1 xClass_2 xClass_3 xClass_4 xClass_5 xClass_6 xClass_7 xClass_8 xClass_9
[1,]        0        1        0        0        0        0        0        0        0
[2,]        0        1        0        0        0        0        0        0        0
[3,]        0        0        0        0        0        1        0        0        0
[4,]        0        1        0        0        0        0        0        0        0
[5,]        0        0        0        0        0        0        0        1        0
[6,]        0        1        0        0        0        0        0        0        0

剩下的就是去掉“x”，但是像这样的事情应该不会太难:

d <- dummy(x,drop = FALSE)
colnames(d) <- sub("x", "", colnames(d))

然后保存到磁盘:

write.csv(d, "somefile.csv", row.names = FALSE)