欢迎来到尧图网

客户服务 关于我们

您的位置:首页 > 房产 > 家装 > 自定义数据集 使用scikit-learn中svm的包实现svm分类

自定义数据集 使用scikit-learn中svm的包实现svm分类

2025/6/16 21:51:03 来源:https://blog.csdn.net/bananapai/article/details/145429451  浏览:    关键词:自定义数据集 使用scikit-learn中svm的包实现svm分类

数据集生成:

 

- 使用  make_classification  函数生成包含1000个样本的数据集,设置20个特征,其中10个是有信息的特征,类别数为2,通过设置  random_state = 42  保证每次运行生成的数据相同。

 

数据划分:

 

- 使用  train_test_split  函数将生成的数据集划分为训练集和测试集,测试集占比为20%,同样通过  random_state = 42  保证划分的一致性。

 

SVM模型:

 

- 初始化  SVC  类,这里使用线性核函数  kernel='linear' 。还有其他核函数可供选择,如  'rbf' (径向基函数核)、 'poly' (多项式核)等,不同的核函数适用于不同的数据分布。

 

- 使用  fit  方法将模型拟合到训练集数据  X_train  和对应的标签  y_train  上。

 

预测与评估:

 

- 使用训练好的模型对测试集  X_test  进行预测,得到预测标签  y_pred 。

 

- 使用  accuracy_score  函数计算预测准确率,评估模型在测试集上的性能。

import numpy as np

from sklearn.datasets import make_classification

from sklearn.model_selection import train_test_split

from sklearn.svm import SVC

from sklearn.metrics import accuracy_score

 

 

# 生成自定义数据集

X, y = make_classification(n_samples=1000, n_features=20, n_informative=10, n_classes=2, random_state=42)

 

# 划分训练集和测试集

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

 

# 初始化并拟合SVM模型

svm_classifier = SVC(kernel='linear')

svm_classifier.fit(X_train, y_train)

 

# 预测

y_pred = svm_classifier.predict(X_test)

 

# 计算准确率

accuracy = accuracy_score(y_test, y_pred)

print(f"Accuracy of SVM classifier: {accuracy}")

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

热搜词