对鸢尾花数据集使用K近邻算法
题目说明
对鸢尾花数据集采用Knn算法,预估在不同的花瓣长宽等条件下,不同的长宽属于三种品种的哪一种。
思路说明
很显然,花瓣的长与宽属于一个平面直角坐标系,对坐标系中的每一个点进行K近邻算法,不同的品种涂上不同的颜色,就得到了一副图,关于三个品种的说明。
源代码
# -*- coding: utf-8 -*-
#为了引入鸢尾花数据集导入的包
from sklearn import datasets
#画点图要用的包
from matplotlib import pyplot
#导入鸢尾花数据
iris = datasets.load_iris()
# 求两点间距离公式的函数,两点为(a,b)与(x,y)
def getLength(a,b,x,y):
length=((a-x)**2+(b-y)**2)