我创建了一个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模型时自动出现的图例。
谢谢 - 我感谢您的帮助!