SVM code example

Inspired from this blog post.

set.seed(1)
library('tidyverse')
library('e1071')

ir <- iris |> 
  mutate(is_setosa = as.factor(ifelse(Species == 'setosa', 'Yes', 'No'))) |> 
  select(-c(Species, Sepal.Length, Petal.Length))

ir <- as.data.frame(ir)

fm_svm <- svm(is_setosa ~ ., data = ir)

plot(fm_svm, data = ir)