Advertisement
Guest User

Untitled

a guest
May 25th, 2019
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.83 KB | None | 0 0
  1. print("Engineering new features...")
  2.  
  3. columns_dataset <- 1:(ncol(data_train)-1);
  4. for (d in columns_dataset) {
  5.  
  6. data_train <- cbind(data_train, data_train[, d] ** 2);
  7. data_test <- cbind(data_test, data_test[, d] ** 2);
  8.  
  9. colnames(data_train)[ncol(data_train)] <- paste("N", d, "squared" , sep = "_");
  10. colnames(data_test)[ncol(data_test)] <- paste("N", d, "squared" , sep = "_");
  11.  
  12. data_train <- cbind(data_train, data_train[, d] ** 3);
  13. data_test <- cbind(data_test, data_test[, d] ** 3);
  14.  
  15. colnames(data_train)[ncol(data_train)] <- paste("N", d, "cube" , sep = "_");
  16. colnames(data_test)[ncol(data_test)] <- paste("N", d, "cube" , sep = "_");
  17.  
  18. # On met la colonne ta rget a la fin
  19. data_train <- data_train[, c(1:(ncol(data_train) - 3), (ncol(data_train) - 1), ncol(data_train), (ncol(data_train) - 2))]
  20. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement