全部版块 我的主页
论坛 数据科学与人工智能 数据分析与数据科学 R语言论坛
1432 3
2017-03-09
我有一个训练数据集trainSetstr(trainSet)
'data.frame':        40091 obs. of  9 variables:
$ Region        : int  2 7 2 5 8 3 9 7 4 3 ...
$ Date          : int  1 1 1 1 1 1 1 1 1 1 ...
$ Weekday       : int  2 2 2 2 2 2 2 2 2 2 ...
$ Apartment.Room: int  1 1 0 1 1 1 1 0 1 1 ...
$ Beds          : int  1 4 1 1 1 1 2 2 1 1 ...
$ Review        : num  4.91 4.93 3.45 3.8 3.02 ...
$ Pic.Quality   : num  0.778 0.569 0.938 0.803 0.984 ...
$ Price         : num  419 511 337 317 280 ...
$ Accept        : int  0 1 0 0 0 1 0 0 1 0 ...

现在采用randomforest预测Accept(是否被接受)
modelofrf <- randomForest(Accept ~ ., data = trainSet, ntree = 2000,
                          importance = TRUE, proximity = FALSE)

但是得到的预测值不是0,1而是
            1             2             3             4             5             6
-2.950523e+00 -1.118509e+00 -3.199625e+00 -2.603652e+00 -2.689188e+00 -1.594579e+00
            7             8            10            11            14            15
-1.794545e+00 -1.851133e+00 -2.312016e-01 -1.639166e+00 -1.504383e+00 -7.678315e-01
           16            17            18            20            21            22
-1.176839e+00 -2.590223e+00 -2.506254e+00  4.390140e-02 -2.916865e+00 -5.731201e-01
           23            24            26            27            29            32
-1.452550e+00 -2.885675e+00 -2.638527e+00 -2.151004e+00 -2.924910e+00 -3.743366e+00

这样的连续数据
我需要自己转换成分类还是我输入的代码有问题?新手一枚,请大神指教。

二维码

扫码加我 拉你入群

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

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

全部回复
2017-3-10 10:41:54
使用逻辑回归同样得到的预测值有正有负
modeloflr <- glm(Accept ~ ., data = trainSet, family = binomial(link='logit'))
            1             2             3             4             5             6
-2.950523e+00 -1.118509e+00 -3.199625e+00 -2.603652e+00 -2.689188e+00 -1.594579e+00
            7             8            10            11            14            15
-1.794545e+00 -1.851133e+00 -2.312016e-01 -1.639166e+00 -1.504383e+00 -7.678315e-01
           16            17            18            20            21            22
-1.176839e+00 -2.590223e+00 -2.506254e+00  4.390140e-02 -2.916865e+00 -5.731201e-01
二维码

扫码加我 拉你入群

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

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

2017-3-10 17:58:27
数据能截图展示下嘛 ?  可以这样做 ifelse(accpet.predict >0.5,1 ,0 )
二维码

扫码加我 拉你入群

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

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

2017-3-14 10:05:49
AlexYoung757 发表于 2017-3-10 17:58
数据能截图展示下嘛 ?  可以这样做 ifelse(accpet.predict >0.5,1 ,0 )
问题已解决,输出预测的时候用语句
pre <- predict(modelofrf, data = trainSet, type = 'response')
将type选择成response即可
多谢指导
二维码

扫码加我 拉你入群

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

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

栏目导航
热门文章
推荐文章

说点什么

分享

扫码加好友,拉您进群
各岗位、行业、专业交流群