能写一个SVM的python代码吗
时间: 2024-03-02 16:25:29 浏览: 63
python实现SVM
当然可以。以下是一个简单的SVM Python代码示例:
```
from sklearn import datasets
from sklearn import svm
# 加载iris数据集
iris = datasets.load_iris()
# 使用前两列特征作为数据
X = iris.data[:, :2]
# 使用后两列特征作为标签
y = iris.target
# 定义一个SVM模型
clf = svm.SVC(kernel='linear')
# 训练模型
clf.fit(X, y)
# 预测新数据
new_data = [[5.0, 3.5]]
print(clf.predict(new_data))
```
这段代码实现了使用SVM对iris数据集进行分类。首先加载iris数据集,使用前两列特征作为数据,后两列特征作为标签。然后定义一个SVM模型,使用线性核函数进行训练,并预测新数据的标签。
阅读全文