Advertisement
krzychuz

Untitled

Jul 20th, 2018
110
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
R 0.92 KB | None | 0 0
  1. library(mlbench)
  2. library(randomForest)
  3. data(BostonHousing2)
  4. dane <- BostonHousing2
  5. dane <- unique(dane)
  6. rows <- nrow(dane)
  7. dane <- dane[sample(rows),]
  8. dane <- na.omit(dane)
  9. n <- 10
  10. div <- rows %/% n
  11. mod <- rows %% n
  12. sampleSizes <- vector(mode = "integer", length = n)
  13. for (i in 1:n){
  14.   if(i<=mod)
  15.     sampleSizes[i] <- div+1
  16.   else
  17.     sampleSizes[i] <- div
  18. }
  19. pred <- data.frame()
  20. test <- data.frame()
  21. for (i in 1:n){
  22.   begin <- sum(sampleSizes[0:(i-1)])
  23.   end <- sum(sampleSizes[1:i])
  24.   trainingset <- dane[c(0:(begin), (end+1):rows),]
  25.   testset <- dane[(begin+1):end,]
  26.   interesujace <- c(2:4, 7:19)
  27.   trainingset <- na.omit(trainingset)
  28.   model <- randomForest(x=trainingset[,interesujace], y=trainingset[,6], ntree=10)
  29.   temp <- as.data.frame(predict(model, testset[,interesujace]))
  30.   pred <- rbind(pred, temp)
  31.   test <- rbind(test, as.data.frame(testset[,6]))
  32. }
  33.  
  34. err <- sqrt(sum((test-pred)^2)/rows)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement