python实现svm二分类代码
时间: 2023-08-06 17:02:58 浏览: 187
下面是一个简单的 Python 代码实现 SVM 二分类:
```python
import numpy as np
from sklearn.svm import SVC
# 创建一个二分类的数据集
X = np.array([[1, 2], [2, 3], [3, 1], [4, 3], [5, 4], [6, 5], [7, 7], [8, 6]])
y = np.array([0, 0, 0, 0, 1, 1, 1, 1])
# 创建一个 SVM 模型
svm = SVC(kernel='linear', C=100)
# 训练模型
svm.fit(X, y)
# 预测结果
print(svm.predict([[4, 5]]))
```
上面的代码中,我们首先创建了一个二分类的数据集,然后使用 scikit-learn 库中的 `SVC` 类创建了一个 SVM 模型,并指定了线性核函数和惩罚参数 C 的值为 100。接着,我们使用 `fit` 方法来训练模型,并使用 `predict` 方法来进行预测。最后,输出预测结果。
相关问题
python实现svm二分类
以下是一个简单的Python实现SVM二分类的示例代码,使用sklearn库中的SVC类:
```
from sklearn import datasets
from sklearn.model_selection import train_test_split
from sklearn import svm
# 加载数据集
iris = datasets.load_iris()
X = iris.data[:, :2] # 选取前两个特征用于二分类
y = iris.target
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=0)
# 创建SVM分类器
clf = svm.SVC(kernel='linear', C=1.0)
# 训练模型
clf.fit(X_train, y_train)
# 预测测试集
y_pred = clf.predict(X_test)
# 输出模型准确率
print("Accuracy:", clf.score(X_test, y_test))
```
在这个示例中,我们使用了Iris数据集,选取前两个特征进行二分类,使用SVM分类器进行训练和预测,并输出模型的准确率。
python svm二分类代码
当使用Python进行SVM(支持向量机)二分类时,可以使用scikit-learn库中的SVC类来实现。下面是一个简单的示例代码:
```python
# 导入所需的库
from sklearn import svm
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
# 准备数据集
# 假设你已经有了特征矩阵X和标签向量y
# X的每一行代表一个样本的特征,y是对应的标签
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建SVM分类器对象
clf = svm.SVC()
# 在训练集上训练模型
clf.fit(X_train, y_train)
# 在测试集上进行预测
y_pred = clf.predict(X_test)
# 计算准确率
accuracy = accuracy_score(y_test, y_pred)
print("准确率:", accuracy)
```
这段代码首先导入了所需的库,然后准备了数据集,包括特征矩阵X和标签向量y。接下来,使用`train_test_split`函数将数据集划分为训练集和测试集。然后,创建了一个SVM分类器对象`clf`,并使用训练集对其进行训练。最后,使用测试集进行预测,并计算准确率。
阅读全文