📜  在Python中创建线性核 SVM(1)

📅  最后修改于: 2023-12-03 15:07:55.460000             🧑  作者: Mango

在Python中创建线性核 SVM

支持向量机(Support Vector Machine,SVM)是一种强大的机器学习算法,可用于分类和回归分析。SVM的目标是在所有可能的决策边界中找到最佳的决策边界,以使决策边界最大化。

在本文中,我们将重点介绍如何在Python中使用线性核创建SVM。

使用scikit-learn库创建线性核 SVM

scikit-learn是Python中一个广泛使用的机器学习库,提供了许多强大的工具和函数,包括SVM。下面是使用scikit-learn库创建线性核SVM的步骤:

  1. 首先,我们需要导入所需的库:
from sklearn.svm import SVC
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
  1. 接下来,我们需要加载数据集。在这个例子中,我们将使用乳腺癌数据集:
data = load_breast_cancer()
X = data.data
y = data.target
  1. 然后,我们将数据集分为训练集和测试集:
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
  1. 接下来,我们可以创建SVM分类器并将其拟合到训练数据中:
svm = SVC(kernel='linear')
svm.fit(X_train, y_train)
  1. 最后,我们可以使用测试数据集进行预测并计算准确性:
y_pred = svm.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print("Accuracy:", accuracy)

从上面的代码中,我们可以看到在Python中使用scikit-learn库创建线性核SVM是非常简单的。

结论

SVM是一种强大的机器学习算法,它可以用于解决许多分类和回归问题。在Python中创建线性核SVM非常简单,只需使用scikit-learn库并遵循上述步骤即可。