0
如何在不破壞目標變量的情況下在插入符號中使用虛擬變量?插入符號虛擬變量不包括目標
set.seed(5)
data <- ISLR::OJ
data<-na.omit(data)
dummies <- dummyVars(Purchase ~ ., data = data)
data2 <- predict(dummies, newdata = data)
split_factor = 0.5
n_samples = nrow(data2)
train_idx <- sample(seq_len(n_samples), size = floor(split_factor * n_samples))
train <- data2[train_idx, ]
test <- data2[-train_idx, ]
modelFit<- train(Purchase~ ., method='lda',preProcess=c('scale', 'center'), data=train)
將失敗,因爲購買變量缺失。如果我將其替換爲data$Purchase <- ifelse(data$Purchase == "CH",1,0)
,則事前插入符號表示這不再是分類而是迴歸問題
你可以做'數據2的
class
$購買< - 數據$購買'事後不能? – mtoto我試過了 - 但這似乎扭曲了矩陣的結果。是否有可能將插入的虛擬變量直接傳入火車?作爲管道? –