Python实现鸢尾花数据集分类问题——基于skearn的SVM
Python实现鸢尾花数据集分类问题——基于skearn的SVM 代码如下: 1 # !/usr/bin/env python 2 # encoding: utf-8 3 __author__ = 'Xiaolin Shen' 4 from sklearn import svm 5 import numpy as np 6 from sklearn import model_selection 7 import matplotlib.pyplot as plt 8 import matplotlib as mpl 9 from matplotlib import colors 10 11 12 13 # 当使用numpy中的loadtxt函数导入该数据集时,假设数据类型dtype为浮点型,但是很明显数据集的第五列的数据类型是字符串并不是浮点型。 14 # 因此需要额外做一个工作,即通过loadtxt()函数中的converters参数将第五列通过转换函数映射成浮点类型的数据。 15 # 首先,我们要写出一个转换函数: 16 # 定义一个函数,将不同类别标签与数字相对应 17 def iris_type(s): 18 class_label={b'Iris-setosa':0,b'Iris-versicolor':1,b'Iris-virginica':2} 19 return