如何使用e1071计算数据点到SVM决策超平面的距离

我使用e1071包创建了一个预测两个类别的线性模型。现在我可以预测类别,但我也想知道每个预测到决策超平面的距离。

以下代码用于对iris数据进行子集划分,创建训练集和预测集:

# 数据应有2个因子,以便只有一个超平面
iris.subset <- subset(iris, iris$Species %in% c("versicolor", "virginica"))
iris.subset$Species <- as.factor(as.character(iris.subset$Species))
# 随机抽样用于训练数据
training.data <- iris.subset[sample(1:nrow(iris.subset), 50, replace=FALSE),]
# 剩余样本构成预测数据
prediction.data <- iris.subset[!(rownames(iris.subset) %in% rownames(training.data)),]

以下代码用于拟合模型:

require(e1071)
svmfit <- svm(Species~., data=training.data, kernel="linear")

从预测集中预测5个样本:

predict(svmfit, prediction.data[sample(1:nrow(prediction.data), 5, replace=FALSE),])

这让我得到了类别。现在,我想要计算这些点到超平面的距离。我该怎么做?

这里有一个类似但未回答的问题,不过是在Matlab中。这里是另一个可能有帮助的页面,但同样是在Matlab中。


回答:

你可以从预测中获取决策值,方法如下:

dd <- prediction.data[sample(1:nrow(prediction.data), 5, replace=FALSE),]
pred <- predict(svmfit, dd, decision.value=T)
pred #        112        139         87        108         70 
#  virginica  virginica versicolor  virginica versicolor 
# attr(,"decision.values")
#     virginica/versicolor
# 112            1.9830355
# 139            0.4160704
# 87            -1.2680673
# 108            2.7181950
# 70            -2.6954507

这些值以属性的形式返回。你可以更直接地访问数据,方法如下:

attr(pred, "decision.values")

这将返回一个值矩阵。

如需更多信息,你可以阅读?predict.svm的帮助页面

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注