Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- print("Engineering new features...")
- columns_dataset <- 1:(ncol(data_train)-1);
- for (d in columns_dataset) {
- data_train <- cbind(data_train, data_train[, d] ** 2);
- data_test <- cbind(data_test, data_test[, d] ** 2);
- colnames(data_train)[ncol(data_train)] <- paste("N", d, "squared" , sep = "_");
- colnames(data_test)[ncol(data_test)] <- paste("N", d, "squared" , sep = "_");
- data_train <- cbind(data_train, data_train[, d] ** 3);
- data_test <- cbind(data_test, data_test[, d] ** 3);
- colnames(data_train)[ncol(data_train)] <- paste("N", d, "cube" , sep = "_");
- colnames(data_test)[ncol(data_test)] <- paste("N", d, "cube" , sep = "_");
- # On met la colonne ta rget a la fin
- data_train <- data_train[, c(1:(ncol(data_train) - 3), (ncol(data_train) - 1), ncol(data_train), (ncol(data_train) - 2))]
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement