sklearn库学习之K-NN算法

最新推荐文章于 2022-11-25 21:33:51 发布

Richard-Tang

最新推荐文章于 2022-11-25 21:33:51 发布

阅读量900

点赞数

分类专栏：大三

本文链接：https://blog.csdn.net/thj19980720/article/details/83066686

版权

本文探讨了Python中sklearn库的K-NN算法，包括k近邻分类和k近邻回归。同时，文章解答了关于代码中函数用法的常见疑惑，并提供了多个与数据可视化、数据预处理和数据生成相关的资源链接，如pandas的plot函数、matplotlib的基础使用、tensorflow的reshape操作以及sklearn的data集介绍。

摘要由CSDN通过智能技术生成

k近邻分类与k近邻回归

import matplotlib.pyplot as plt
from sklearn.neighbors import KNeighborsRegressor
from sklearn.neighbors import KNeighborsClassifier
from sklearn.model_selection import train_test_split
import mglearn
import numpy as np
#############
X,y = mglearn.datasets.make_forge()
X_train,X_test,y_train,y_test = train_test_split(X,y,random_state = 0)
clf = KNeighborsClassifier(n_neighbors = 3)
clf.fit(X_train,y_train)

print("Test set predictions:{}".format(clf.predict(X_test)))
print("Test set accuracy:{:.2f}".format(clf.score(X_test,y_test)))

fig, axes = plt.subplots(1,3,figsize = (10,3))
for n_neighbors,ax in zip([1,3,9],axes):
    clf = KNeighborsClassifier(n_neighbors = n_neighbors).fit(X,y)
    #画图，决策边界可视化
    mglearn.plots.plot_2d_separator(clf,X,fill = True, eps = 0.5,ax = ax, alpha = 0.4)
    mglearn.discrete_scatter(X[:,0],X[:,1],y,ax = ax)#标点
    
    ax.set_