python支持向量机代码
时间: 2023-07-22 17:10:10 浏览: 113
下面是一个简单的 Python 支持向量机(SVM)的实现代码:
```python
from sklearn import svm
import numpy as np
# 准备数据
X = np.array([[0, 0], [1, 1]])
y = np.array([0, 1])
# 创建 SVM 模型对象
clf = svm.SVC(kernel='linear')
# 训练 SVM 模型
clf.fit(X, y)
# 预测数据
print(clf.predict([[2., 2.]]))
```
这个例子中,我们准备了一个二维的数据集,包含两个类别,然后创建了一个使用线性核的 SVM 模型对象,并将数据集用于训练该模型。最后,我们使用模型对一个新数据([2, 2])进行预测,并输出预测结果。
阅读全文