支持向量机python代码
时间: 2023-07-22 10:43:37 浏览: 87
以下是一个简单的支持向量机的Python代码示例:
```python
from sklearn import svm
import numpy as np
# 训练数据
X_train = np.array([[0, 0], [1, 1]])
y_train = np.array([0, 1])
# 创建一个支持向量机分类器
clf = svm.SVC(kernel='linear')
# 训练分类器
clf.fit(X_train, y_train)
# 预测
print(clf.predict([[2., 2.]]))
```
在这个例子中,我们使用了`SVC`类来创建一个线性支持向量机分类器,并使用`fit`方法来训练分类器。最后,我们使用`predict`方法来预测一个新的样本的类别。
阅读全文