1. 引言
支持向量机(Support Vector Machine,SVM)是一种强大的监督学习算法,常用于分类和回归任务。在Python中,我们可以使用scikit-learn库来实现SVM分类器。以下是一个简单的SVM分类器实现示例。
2. 准备数据
首先,我们需要一些数据来训练和测试我们的SVM分类器。这里我们使用scikit-learn库中的鸢尾花(Iris)数据集。
from sklearn import datasets
# 加载数据集
iris = datasets.load_iris()
X = iris.data
y = iris.target
3. 划分数据集
为了训练和测试模型,我们需要将数据集划分为训练集和测试集。
from sklearn.model_selection import train_test_split
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
4. 创建SVM分类器
接下来,我们创建一个SVM分类器实例,并使用训练集进行训练。
from sklearn import svm
# 创建SVM分类器实例
clf = svm.SVC(kernel='linear')
# 使用训练集训练模型
clf.fit(X_train, y_train)
5. 评估模型
使用测试集评估模型的性能。
from sklearn.metrics import accuracy_score
# 使用测试集评估模型
y_pred = clf.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print(f"模型准确率:{accuracy:.2f}")
6. 预测新数据
最后,我们可以使用训练好的模型来预测新数据。
# 预测新数据
new_data = [[5.1, 3.5, 1.4, 0.2]]
prediction = clf.predict(new_data)
print(f"预测结果:{prediction[0]}")
7. 总结
以上就是一个简单的SVM分类器实现示例。在实际应用中,您可能需要调整参数、选择不同的核函数或进行特征工程以提高模型的性能。希望这个示例能帮助您更好地理解SVM分类器在Python中的实现。