无法在预报。部分中设置type =“ class” >> [

问题描述 投票:0回答:1
我正在尝试对数据集使用决策树分类,该数据集包含2个要素和1个因变量,看起来像:购买的年龄工资(是/否)

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 90.03296703 0.03296703 0.03296703 0.03296703

我需要获得的不是概率,而是最可能的结果。但是当我使用

y_pred = predict(classifier, newdata = test_set[-3], type = 'class')

我知道

predict.rpart(classifier,newdata = test_set [-3],类型=“ class”):“ rpart”对象的无效预测

您能帮我吗?

[我正在尝试在我的数据集中使用决策树分类,该数据集包含2个要素和1个因变量,看起来像:购买的年龄工资(是/否)26 43000 0 17 57000 0 19 76000 0 ...

r machine-learning predict rpart
1个回答
0
投票
© www.soinside.com 2019 - 2024. All rights reserved.