机器学习——KNN回归

1、前提知识:

  • 回归:可以理解为拟合,就是根据训练数据的趋势,对输入数据进行预测。
  • KNN回归:是一种有监督学习,因为需要提供目标数据(target)

2、案例:

  • 用KNN回归拟合sin函数,首先通过sin函数上的点训练KNN模型,然后用随机生成的点拟合函数。
  • 代码:
# 首先随机生成100个数用于sin函数生产成训练数据
import math
import numpy as np
x_train = np.random.random(100)*10
y_train = np.sin(x_train)
x_train.shape, y_train.shape
# 转换训练数据维度:x为二维,y为一维
x_train = x_train.reshape(100,-1) # -1本身代替1,目的是简化参数输入的计算
x_train.shape
# 进行KNN回归
from sklearn.neighbors import KNeighborsRegressor
knn = KNeighborsRegressor()
knn.fit(x_train,y_train)
# 进行KNN预测
x_test = np.linspace(0,10,100).reshape(100,1)
x_test.shape
y_test = knn.predict(x_test)
# 绘图查看
import matplotlib.pyplot as plt
plt.scatter(x_train,y_train,c='blue')
plt.plot(x_test,y_test,c='red')

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/sz1125218970/article/details/132533127