2014-05-11 2 views
0

분류를 위해 임의의 포레스트를 사용하고 싶습니다만, 해결책을 찾을 수없는 두 가지가 있습니다. 첫 번째는 어떻게 첨부 할 수 있습니까? 예측 결과는 데이터 프레임으로 전달됩니다. 둘째, 예측 품질에 대한 우발적 인 표 이외의 통찰력 (차트, 통계 등)을 어떻게 얻을 수 있습니까? 내 데이터 프레임 예는 다음과 같습니다.랜덤 포레스트에서 - 어떻게 데이터 프레임에 예측 결과를 첨부 할 수 있습니까?

dat <- read.table(text = " TargetVar Var1 Var2  Var3 
0  0  0   7 
0  0  1   1 
0  1  0   3 
0  1  1   7 
1  0  0   5 
1  0  1   1 
1  1  0   0 
1  1  1   6 
0  0  0   8 
0  0  1   5 
1  1  1   4 
0  0  1   2 
1  0  0   9 
1  1  1   2 ", header = TRUE) 

dat$dat$TargetVar <- as.factor(dat$TargetVar) #changed my target variable into a factor 
rf <- randomForest(TargetVar~Var1 + Var2+ Var3 ,data=dat,ntree=10,importance=TRUE) #produced the model 
table(predict(rf),dat$TargetVar) # got the contingency table 

감사합니다.

답변

2

질문의 첫 부분 만 대답 할 수 있습니다.

predict(rf) 
: 당신의 예측의 결과를 고려

: 나는 데이터 프레임

에 예측 결과를 첨부 할 수있는 방법을

는 사용할 수에게 cbind 기능을이 작업을 수행하려면

데이터 프레임으로 변환

predResults <- data.frame(predict(rf)) 

그리고이 결과를 포함하는 원래의 데이터 프레임 (DAT)를 업데이트 :

dat <- cbind(dat,predResults) 

> dat 
     TargetVar Var1 Var2 Var3 predict.rf. 
    1  0  0 0 7 0.40000000 
    2  0  0 1 1 0.69642857 
    3  0  1 0 3 1.00000000 
    4  0  1 1 7 0.50000000 
    5  1  0 0 5 0.30000000 
    6  1  0 1 1 0.22380952 
    7  1  1 0 0 0.47817460 
    8  1  1 1 6 0.52916667 
    9  0  0 0 8 0.44444444 
    10  0  0 1 5 0.07142857 
    11  1  1 1 4 0.66666667 
    12  0  0 1 2 1.00000000 
    13  1  0 0 9 0.15666667 
    14  1  1 1 2 0.51785714 
+0

안녕, 내 대답을 편집했다. 예측을 가져와 데이터 프레임으로 변환하면 cbind를 사용하여 예측을 원래 데이터 프레임에 추가 할 수 있습니다. – cbento

+0

안녕하세요, 감사합니다 .. – mql4beginner