# 导包
import numpy as np
import matplotlib.pyplot as plt
下面的是样本的数据集和对应的标记(使用数字0,1进行区别表示)
下面的这个数据特征,实际上就是我们二维坐标平面上面的数据点罢了
# 样本特征
data_X = [
[1.3, 6],
[3.5, 5],
[4.2, 2],
[5, 3.3],
[2, 9],
[5, 7.5],
[7.2, 4 ],
[8.1, 8],
[9, 2.5]
]
下面的这个就是标记:0,1表示两类的数据
# 样本标记
data_y = [0,0,0,0,1,1,1,1,1]
使用下面的这个np.array把我们的列表类型转换为数组:
# 训练集
X_train = np.array(data_X)
y_train = np.array(data_y)
为了清晰看到这个具体内容,我们可以看看这个转换之后的结果:
我们也可以采用下面的这个方法,调用plt绘图方法,把这个数据可视化展现出来:
使用散点图的方式进行绘制,不同的标记使用不同的图样进行绘制方便我们进行不同的标记之间的区分;调用plt.show()把这个图像展示出来;
上面的这个是给定的数据,我们如何使用散点图表示不同的标记点之间的关系,如果新给了一点,也就是新的样本点,我们如何把这个新的样本点添加到我们的这个散点图里面去呢?
可以使用下面的这个方法,分别获取这个新的样本点的横坐标和纵坐标,使用新的marker进行标记即可;
首先我们需要知道,之前的10个数据的特征都是已知的,我们需要做的就是把这个位置的样本点进行标记,确定他是属于哪一类里面的;
因此我们需要计算这个新的样本点和原始数据之间的距离:
在计算出来的距离的基础上面,对于数据进行排序:
接下来就是确定这个k值(不清楚的伙伴可以去看之前的那个理论介绍的文章,k就是我们以几个数据点作为依据),当然这个k的选择是有讲究的,但是现在我们暂时不深究,假设k=5(随机选择的);
确定这k个参考点里面的不同类别的个数:即0有几个,1有几个;
进行预测:这个里面的23后面有两个[0],这个里面的第一个表示取出来0下标的元组,第二个0表示取出来这个元组里面的第一个数字,实际上这个里面(22行打印的结果里面)只有一个元组
导包,传入k值(对于创建的实例传参)
参数是二维数组,我们需要的是一维数组,所以需要使用这个reshape方法进行维度的转换,然后进行预测,是不是很快?得到的结果是一样的