r - C5.0模型需要一个结果要素

Question

我正在使用credit.csv来构建学习树，该数据可在以下位置找到:

https://github.com/stedy/Machine-Learning-with-R-datasets/blob/master/credit.csv

并且我已经执行了以下步骤:

credit<-read.csv("credit.csv")
set.seed(12345)
credit_rand<-credit[order(runif(1000)),]
credit_train<-credit_rand[1:900,]
credit_test<-credit_rand[901:1000,]
library(C50)
credit_model<-C5.0(credit_train[-21],credit_train$default)

在我遵循的指南中，我似乎应该摆脱默认值的最后一列，但出现以下错误:

Error en C5.0.default(credit_train[, -21], credit_train$default) : 
  C5.0 models require a factor outcome

我尝试将最后一行更改为:

credit_model<-C5.0(credit_train[,-21],credit_train$default)

但根本没有成功。

有什么帮助吗？

Answer 1

您的问题是C5.0 models require a factor outcome。您将结果指定为credit_train$default，它是1/2的结果，但R已将其读取为数字而不是因数:

str(credit_train$default)
int [1:900] 2 1 1 1 2 1 2 2 1 1 ...

然后的解决方案是将其转换为一个因子:

credit_train$default<-as.factor(credit_train$default)
str(credit_train$default)

Factor w/ 2 levels "1","2": 2 1 1 1 2 1 2 2 1 1 ...

然后运行您的训练:

 credit_model<-C5.0(credit_train[-21],credit_train$default)