SVM支持向量机分类Python代码示例
时间: 2023-10-07 08:09:26 浏览: 93
svm支持向量机python 代码示例
当然,我可以为您提供SVM支持向量机分类Python代码示例。以下是一个简单的示例:
```python
from sklearn import datasets
from sklearn import svm
# 加载 Iris 数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
# 训练模型
clf = svm.SVC()
clf.fit(X, y)
# 预测新的数据点
new_data = [[5.1, 3.5, 1.4, 0.2], [6.2, 3.4, 5.4, 2.3]]
print(clf.predict(new_data))
```
在这个示例中,我们使用 Scikit-Learn 库中的 SVM 模型进行分类。我们加载 Iris 数据集作为我们的训练数据,然后训练模型并进行预测。我们输入两个新的数据点,然后预测它们的类别。SVM 模型的输出结果是一个 numpy 数组,其中包含了每个新数据点的预测类别。
希望这个示例能够帮到您,如果还有其他问题,随时问我!
阅读全文