用R的nnet包做分类，predict函数出错，求大神

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

加上type=“class”就出错，代码如下：

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏0 回帖

关键词：predict Pred ICT edi dic

没人吗

1. library(tidyverse)
   
2. library(caret)
   
3. 
   
4. data(iris)
   
5. iris %>% glimpse # 查看数据集
   
6. 
   
7. # 划分训练集与测试集
   
8. iris.idx <- createDataPartition(iris$Species, p = 0.8, list = F)
   
9. iris.tr <- iris[iris.idx, ]
   
10. iris.ts <- iris[-iris.idx, ]
    
11. 
    
12. library(nnet)
    
13. iris.nnet <- nnet(
    
14.   Species ~ .,
    
15.   iris.tr,
    
16.   size = 20,
    
17.   linout = T,
    
18.   maxit = 1e5,
    
19.   MaxNWts = 1e5,
    
20.   abstol = 0.001
    
21. )
    
22. 
    
23. # 绘制神经网络
    
24. library(NeuralNetTools)
    
25. iris.nnet %>% plotnet(
    
26.   circle_cex = 3,
    
27.   cex_val = 0.7,
    
28.   alpha = 0.7,
    
29.   circle_col = "gold",
    
30.   pos_col = "red3",
    
31.   neg_col = "dodgerblue",
    
32.   max_sp = T
    
33. )
    
34. 
    
35. # 预测测试集并输出混淆矩阵
    
36. iris.nnet %>%
    
37.   predict(iris.ts, type = "class") %>%
    
38.   table(iris.ts$Species) %>%
    
39.   caret::confusionMatrix(mode = "everything")

复制代码

narcissism0923 发表于 2018-9-25 13:02

谢谢，可我想知道为啥predict会报错呢，我在predict里面加上softmax=TRUE，就可以了，不知道为什么