如何在R中绘制SVM的分类图

Question

我在R中有一个SVM,我现在想绘制这台机器的分类空间.我在互联网上找到了一些例子,但我似乎无法理解它们.

我的R脚本如下:

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)

我无法让plot命令工作.我想要一个像这样的图表http://bm2.genes.nig.ac.jp/RGM2/R_current/library/e1071/man/images/plot.svm_001.png

Answer 1

首先,该plot.svm函数假设数据在两个维度上变化.您在示例中使用的数据只是一维的,因此必须在一条线上绘制决策边界,这是不受支持的.其次,函数似乎需要一个数据框作为输入,你正在使用向量.

这应该工作......

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

或者,您可以使用kernlab包:

library(kernlab)

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