SVM支持向量机python代码 四分类
时间: 2023-09-11 14:09:05 浏览: 134
SVM支持向量机多分类python
5星 · 资源好评率100%
下面是一个简单的四分类SVM支持向量机的Python代码示例:
```python
from sklearn import svm
import numpy as np
# 训练数据
X_train = np.array([[0, 0], [1, 1], [1, 0], [0, 1], [2, 2], [3, 3], [2, 3], [3, 2]])
y_train = np.array([0, 0, 1, 1, 2, 2, 3, 3])
# 创建SVM分类器
clf = svm.SVC(kernel='linear', C=1)
# 训练模型
clf.fit(X_train, y_train)
# 测试数据
X_test = np.array([[1.5, 1.5], [2.5, 2.5], [0.5, 0.5], [2.5, 1.5]])
y_test = np.array([0, 2, 1, 3])
# 预测
y_pred = clf.predict(X_test)
# 输出预测结果
print("预测结果:", y_pred)
# 输出准确率
accuracy = clf.score(X_test, y_test)
print("准确率:", accuracy)
```
在这个例子中,我们使用了线性核函数,设置正则化参数C为1。我们训练了一个四分类的SVM模型,并使用测试数据对其进行了验证,输出了预测结果和准确率。
阅读全文