创建KNN类
由于数据集在二维平面内,所以在类的初始化函数中创造数据集的横纵坐标以及对应的标签。
class KNNDeal:
def init(self):
self.module = "这是绘制KNN的决策边界代码"
self.raw_data_X = [[3.85, 3.05],
[1.57, 2.63],
[4.28, 4.42],
[5.68, 3.64],
[1.94, 3.68],
[2.49, 2.66],
[0.49, 3.66],
[0.12, 1.12],
[2.25, 2.04],
[0.23, 0.54],
[1.33, 1.49],
[2.35, 0.09],
[1.46, 1.63],
[3.66, 3.93],
[5.11, 0.39],
[5.69, 1.14],
[4.03, 2.49],
[4.92, 1.62],
[5.26, 2.05],
[4.26, 2.05],
[5.84, 1.31]
]
self.raw_data_Y = [0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2];
self.x_train = np.array(self.raw_data_X)
self.y_train = np.array(self.raw_data_Y)