KNN 回归模型

好长一段时间一直以为 k 近邻模型就是做分类用的，其实还可以做回归任务。借助 mglearn 的示例可以很好的理解。

1、k = 1

import mglearn
import matplotlib.pyplot as plt

mglearn.plots.plot_knn_regression(n_neighbors=1)

plt.show()

 

2、k = 2

import mglearn
import matplotlib.pyplot as plt

mglearn.plots.plot_knn_regression(n_neighbors=2)

plt.show()

 

3、 k = 5

import mglearn
import matplotlib.pyplot as plt

mglearn.plots.plot_knn_regression(n_neighbors=5)

plt.show()

 

 使用 k 近邻进行回归时，预测结果为 k 个邻居的平均值。