【发布时间】：2020-07-20 08:05:48
【问题描述】：
我正在尝试对我的数据集使用决策树分类，该数据集包含 2 个特征和 1 个因变量，如下所示： 工资购买年龄（Y/N）
26 43000 0
17 57000 0
19 76000 0
27 58000 0
27 84000 0
32 150000 1
25 33000 0
如果我使用
classifier = rpart(formula = Purchased ~ ., data = training_set)
我得到类似的结果
2 4 5 9
0.03296703 0.03296703 0.03296703 0.03296703
我需要的不是概率，而是最可能的结果。但是当我使用
y_pred = predict(classifier, newdata = test_set[-3], type = 'class')
我明白了
predict.rpart 中的错误（分类器，newdata = test_set[-3]，type = "class") : "rpart" 对象的预测无效
你能帮我解决这个问题吗？
【问题讨论】：
-
class(training_set$Purchased)能得到什么？ -
class(training_set$Purchased) 返回整数
标签： r machine-learning predict rpart