如何在SVM图上删除自动创建的图例

时间:2015-02-27 15:18:17

标签: svm legend

我创建了一个SVM模型,并且还成功地绘制了R2中的轮廓线。请参阅下面的示例,其中我定义了两个变量" x"和" y"这将用于确定"指标"的分类。使用SVM。训练集是原始数据的50%,为简单起见,测试集是原始数据的另外50%。

我的问题在于SVM情节。 我只是希望能够摆脱(或可能移动)出现在我的情节右侧的结果图例。当我尝试在此图上绘制其他点时,它会因缩放而混乱上。

#simulate the data, x and y are used to predict indicator
x = c(rep(1,10), rep(10,10), rep(20,10))
y = c(rep(2,10), rep(12,10), rep(24,10))
indicator = c(rep("low", 10), rep("med",10), rep("high",10))

df = data.frame(x,y,as.factor(indicator))

# create train and test sets, 50% of data each
set.seed(123)
end = length(df$x)
i = 1:end
train = sample(i, (.5 * end), replace = FALSE)
train = sort(train)
test = setdiff(i,train)

TRAINSET = df[train,]
TESTSET = df[test,]

# use SVM to create a predictive model
svm.model = svm(TRAINSET[,3] ~ ., data = TRAINSET[,c(1,2)], cost = 1000, gamma = .5)

# plot SVM
plot(svm.model, TRAINSET)

我可以使用legend()函数添加图例但我无法弄清楚如何删除在绘制SVM模型时自动出现的图例。

谢谢 - 我感谢您的帮助!

0 个答案:

没有答案