pythons实现支持向量机
时间: 2023-11-15 10:53:32 浏览: 81
基于Python实现支持向量机【100011026】
一个在Python中实现支持向量机(SVM)的常用库是scikit-learn。你可以使用scikit-learn库中的svm模块来实现SVM算法。下面是一个简单的示例代码来展示如何在Python中使用scikit-learn实现SVM:
```python
from sklearn import svm
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
# 加载数据
iris = load_iris()
X = iris.data
y = iris.target
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)
# 创建SVM分类器
clf = svm.SVC(kernel='linear')
# 训练模型
clf.fit(X_train, y_train)
# 在测试集上进行预测
y_pred = clf.predict(X_test)
```
以上代码演示了如何使用scikit-learn库中的svm.SVC类来创建一个SVM分类器,并使用iris数据集进行训练和测试。你可以根据自己的需求调整代码中的参数和数据。
阅读全文