Guest User

Untitled

a guest
Jan 18th, 2018
94
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.39 KB | None | 0 0
  1. ctrl <- trainControl(method = "repeatedcv",
  2. number = 10,
  3. repeats = 10,
  4. search = "grid",
  5. returnData = TRUE,
  6. returnResamp = "final",
  7. savePredictions = "all",
  8. classProbs = TRUE,
  9. sampling = "smote",
  10. summaryFunction = twoClassSummary,
  11. selectionFunction = "best",
  12. allowParallel = TRUE)
  13.  
  14. library(caret)
  15. library(DMwR)
  16.  
  17. set.seed(2969)
  18. imbal_train <- twoClassSim(10000, intercept = -20, linearVars = 20)
  19. table(imbal_train$Class)
  20.  
  21. set.seed(9560)
  22. smote_train <- SMOTE(Class ~ ., data = imbal_train)
  23. table(smote_train$Class)[2]
  24.  
  25. perc.over <- 100*(table(smote_train$Class)[2]-table(imbal_train$Class)[2])/table(imbal_train$Class)[2]
  26. perc.under <- 100*table(smote_train$Class)[1]/(table(smote_train$Class)[2]-table(imbal_train$Class)[2])
  27.  
  28. set.seed(1234)
  29. imbal_train <- twoClassSim(10000, intercept = -40, linearVars = 40)
  30. table(imbal_train$Class)
  31.  
  32. set.seed(5678)
  33. smote_train <- SMOTE(Class ~ ., data = imbal_train)
  34. table(smote_train$Class)
  35.  
  36. perc.over <- 100*(table(smote_train$Class)[2]-table(imbal_train$Class)[2])/table(imbal_train$Class)[2]
  37. perc.under <- 100*table(smote_train$Class)[1]/(table(smote_train$Class)[2]-table(imbal_train$Class)[2])
Add Comment
Please, Sign In to add comment