机器学习基础（四）之KNN算法简析

最新推荐文章于 2024-04-09 07:19:51 发布

不安分的coder

最新推荐文章于 2024-04-09 07:19:51 发布

阅读量2.7k

点赞数 7

分类专栏： marchine_learing

本文链接：https://blog.csdn.net/qq_37982109/article/details/87896497

版权

Jupyter Notebook

k近邻算法

import numpy as np
from matplotlib import pyplot as plt
import math

#自己弄的训练集和结果
X_train = np.array([[1,1.2],[3,3.3],[5,2.2],[7,5.7],[9,8.4]])
X_train
array([[1. , 1.2],
       [3. , 3.3],
       [5. , 2.2],
       [7. , 5.7],
       [9. , 8.4]])
       
Y_train = np.array([0,0,0,1,1])

Y_train
array([0, 0, 0, 1, 1])

plt.scatter(x_train[y_train == 0,0],x_train[y_train == 0,1],color="r",label="normal")
plt.scatter(x_train[y_train == 1,0],x_train[y_train == 1,1],color="g",label="bad")
plt.title("search")


`#x是我们要测试的值在下图用蓝色表示
x = np.array([6,3.7])

b
plt.scatter(x_train[y_train == 0,0],x_train[y_train == 0,1],color="r",label="normal")
plt.scatter(x_train[y_train == 1,