PYTHON机器学习——KNN(k近邻算法)
代码实现
首先,导入需要用到的包
import numpy as np
import matplotlib.pyplot as plt
初始化模拟数据集和对应的分类
data_x=[[1.15,2.34],
[1.55,2.47],
[3.78,4.98],
[3.34,4.56],
[2.22,2.34]]
data_y=[1,1,0,0,1]
X_train=np.array(data_x)
Y_train=np.array(data_y)
画图看一下两类数据在图上的分布情况
1