Python实现支持向量机(SVM)算法

上传:theme_93669 浏览: 60 推荐: 0 文件:py 大小:694B 上传时间:2023-11-26 05:30:25 版权申诉

在机器学习领域,支持向量机(SVM)是一种强大的分类和回归算法。本文将介绍如何使用Python编写支持向量机的代码实现。支持向量机通过找到数据中的支持向量,构建一个能够最好地将不同类别分开的超平面,从而实现高效的分类。在Python中,我们可以使用scikit-learn库来轻松实现支持向量机。以下是一个简单的示例代码,演示了如何在Python中使用支持向量机进行分类:

# 导入必要的库
from sklearn import svm
from sklearn.model_selection import train_test_split
from sklearn.datasets import make_classification

# 生成模拟数据
X, y = make_classification(n_samples=100, n_features=2, n_informative=2, n_redundant=0, random_state=42)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

# 创建支持向量机模型
model = svm.SVC()

# 训练模型
model.fit(X_train, y_train)

# 在测试集上进行预测
predictions = model.predict(X_test)

# 输出预测结果
print(predictions)

以上代码简单地演示了如何使用scikit-learn库中的SVC类创建和训练支持向量机模型,并在测试集上进行预测。通过调整参数和使用不同的数据集,你可以进一步定制和优化你的支持向量机模型。

上传资源
用户评论