Comment tracer un graphe de classification d'un SVM dans R

Question

J'ai un SVM dans R et je voudrais maintenant tracer l'espace de classification pour cette machine. J'ai trouvé quelques exemples sur Internet, mais je n'arrive pas à les comprendre.

Mon script R est la suivante:

library(e1071) 
day_of_week <- c(0,1,2,3,4,5,6) 
holiday <- factor(c(T, F, F, F, F, F, T)) 
model <- svm(day_of_week, holiday) 
plot(model, day_of_week, holiday) 

je ne peux pas obtenir la commande de complot pour travailler. Je voudrais un graphique quelque chose comme ça http://bm2.genes.nig.ac.jp/RGM2/R_current/library/e1071/man/images/plot.svm_001.png

Answer 1

Tout d'abord, la fonction plot.svm suppose que les données varient selon deux dimensions. Les données que vous avez utilisées dans votre exemple sont uniquement unidimensionnelles et la limite de décision doit donc être tracée sur une ligne, ce qui n'est pas pris en charge. Deuxièmement, la fonction semble avoir besoin d'une trame de données en entrée et vous travaillez avec des vecteurs.

Cela devrait fonctionner ...

library(e1071) 

day = c(0,1,2,3,4,5,6) 
weather = c(1,0,0,0,0,0,0) 
happy = factor(c(T,F,F,F,F,F,F)) 

d = data.frame(day=day, weather=weather, happy=happy) 
model = svm(happy ~ day + weather, data = d) 
plot(model, d) 

Answer 2

Vous pouvez également utiliser le package kernlab:

library(kernlab) 

model.ksvm = ksvm(happy ~ day + weather, data = d, type="C-svc") 
plot(model.ksvm, data=d)