Как выбрать два желаемых столбца в dataframe для построения модели SVM?

Question

У меня есть фрейм данных с несколькими столбцами, и я хочу построить модель svm.Вопрос в том, как выбрать два конкретных столбца в качестве данных.Например, если мой фрейм данных выглядит следующим образом:

df <- data.frame("x1"=c(1,2,3),"x2"=c(4,5,6),"x3"=c(7,8,9),"x4"=c(1,0,1))

Я хочу построить модель svm с x2 и x3 относительно x4.

Answer 1

Это должно сработать:

library(e1071)
df <- data.frame("x1"=c(1,2,3),"x2"=c(4,5,6),"x3"=c(7,8,9),"x4"=c(1,0,1))

#SVM model
model <- svm(x4 ~ x2 + x3, data = df)
summary(model)

#Model predictions
pred <- predict(model, subset(df,select=c(x2,x3)))
print(pred)

В частности, я думаю, что функция subset - это то, что вы ищете: subset(df,select=c(x2,x3))

Answer 2

Вы можете использовать select из dplyr, например:

df <- select(df, x1,x2,x3)