将一个.csv文件有多组多维数据,第一列是标签名,利用SVM进行分类的机器学习代码,预测最后一列数据属于哪一类,最后可视化展现各类分类情况效果

时间: 2023-12-23 18:01:54 浏览: 23
好的,以下是一个简单的Python代码示例,使用SVM对多组多维数据进行分类,并预测最后一列数据属于哪一类,最后使用matplotlib库可视化展现各类分类情况效果: ``` python import pandas as pd import numpy as np from sklearn import svm from sklearn.preprocessing import LabelEncoder import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D # 加载数据 data = pd.read_csv('data.csv') x = data.iloc[:, :-1].values y = data.iloc[:, 0].values predict_x = data.iloc[:, -1].values.reshape(-1, 1) # 将标签名编码为数字 le = LabelEncoder() y = le.fit_transform(y) # 训练SVM模型 clf = svm.SVC(kernel='linear', C=1) clf.fit(x, y) # 预测分类结果 y_pred = clf.predict(x) predict_y = clf.predict(predict_x) # 可视化分类结果 fig = plt.figure() if x.shape[1] == 2: # 二维平面展示 plt.scatter(x[:, 0], x[:, 1], c=y_pred) plt.scatter(predict_x[:, 0], np.zeros_like(predict_y), c=predict_y, marker='x', s=1000) plt.xlabel('Feature 1') plt.ylabel('Feature 2') else: # 三维空间展示 ax = fig.add_subplot(111, projection='3d') ax.scatter(x[:, 0], x[:, 1], x[:, 2], c=y_pred) ax.scatter(predict_x[:, 0], np.zeros_like(predict_y), np.zeros_like(predict_y), c=predict_y, marker='x', s=1000) ax.set_xlabel('Feature 1') ax.set_ylabel('Feature 2') ax.set_zlabel('Feature 3') plt.show() ``` 上述代码中,我们首先使用pandas库加载名为"data.csv"的.csv文件,然后将数据集分为特征数据和标签数据。由于标签名是字符串类型的,我们需要将其编码为数字类型的标签。接下来,我们使用SVM训练模型,并使用训练好的模型对特征数据进行分类预测。最后,我们使用matplotlib库将分类结果可视化展示。如果特征数据为二维数据,则在图表中展示二维平面,否则在三维空间中展示。在图表中,不同标签名对应的数据点颜色不同,以显示各类分类情况。同时,我们用红色的叉号(x)标注出预测数据点的位置,并将其分类结果用红色或蓝色的颜色表示,以便比较预测结果和实际分类情况的差异。

相关推荐

最新推荐

recommend-type

python,sklearn,svm,遥感数据分类,代码实例

对于分类,SVM最初用于解决二分类问题,多分类问题可通过构建多个SVM分类器解决。SVM具有两大特点:1.寻求最优分类边界,即求解出能够正确划分训练数据集并且几何间隔最大的分离超平面,这是SVM的基本思想;2.基于核...
recommend-type

机器学习分类算法实验报告.docx

对于KNN,SVM,adaboost以及决策树等分类算法对数据集运行结果进行总结,代码点我博文
recommend-type

基于多分类非线性SVM(+交叉验证法)的MNIST手写数据集训练(无框架)算法

1.将数据进行预处理。 2.通过一对一方法将45类训练样本((0,1),(0,2),…(1,2)…(2,3))送入交叉验证法,训练算法为smo 3.得出45个模型,测试时在利用投票法判定 数据结构 '''**************************************...
recommend-type

z-blog模板网站导航网站源码 带后台管理.rar

z-blog模板网站导航网站源码 带后台管理.rarz-blog模板网站导航网站源码 带后台管理.rar
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

list根据id查询pid 然后依次获取到所有的子节点数据

可以使用递归的方式来实现根据id查询pid并获取所有子节点数据。具体实现可以参考以下代码: ``` def get_children_nodes(nodes, parent_id): children = [] for node in nodes: if node['pid'] == parent_id: node['children'] = get_children_nodes(nodes, node['id']) children.append(node) return children # 测试数
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。
recommend-type

"互动学习:行动中的多样性与论文攻读经历"

多样性她- 事实上SCI NCES你的时间表ECOLEDO C Tora SC和NCESPOUR l’Ingén学习互动,互动学习以行动为中心的强化学习学会互动,互动学习,以行动为中心的强化学习计算机科学博士论文于2021年9月28日在Villeneuve d'Asq公开支持马修·瑟林评审团主席法布里斯·勒菲弗尔阿维尼翁大学教授论文指导奥利维尔·皮耶昆谷歌研究教授:智囊团论文联合主任菲利普·普雷教授,大学。里尔/CRISTAL/因里亚报告员奥利维耶·西格德索邦大学报告员卢多维奇·德诺耶教授,Facebook /索邦大学审查员越南圣迈IMT Atlantic高级讲师邀请弗洛里安·斯特鲁布博士,Deepmind对于那些及时看到自己错误的人...3谢谢你首先,我要感谢我的两位博士生导师Olivier和Philippe。奥利维尔,"站在巨人的肩膀上"这句话对你来说完全有意义了。从科学上讲,你知道在这篇论文的(许多)错误中,你是我可以依