其他分享
首页 > 其他分享> > (机器学习)多项式回归 (解决欠拟合问题)

(机器学习)多项式回归 (解决欠拟合问题)

作者:互联网

多项式回归

面对问题

欠拟合

在训练集与测试集都不能获得很好的拟合数据时,认为该假设出现了欠拟合(模型过于简单)

过拟合

在训练集上能获得较好拟合,在训练集以外的数据集上却不能很好的拟合数据

欠拟合

多项式回归

PolynomialFeatures

from sklearn.preprocessing import PolynomialFeatures

c = [[5, 10]]
p1 = PolynomialFeatures()
b = p1.fit_transform(c)
print(b)


[[  1.   5.  10.  25.  50. 100.]]

Process finished with exit code 0

使用PolynomialFeatures

from sklearn.preprocessing import PolynomialFeatures
from sklearn.linear_model import LinearRegression
import numpy as np
import matplotlib.pyplot as plt


# 样本的训练数据,特征和目标值
x_train = [[6], [8], [10], [14], [18]]  # 大小
y_train = [[7], [9], [13], [17.5], [18]]  # 大小
# 使用PolynomialFeatures
poly2 = PolynomialFeatures(degree=2)
x_train_poly2 = poly2.fit_transform(x_train)
# 一次线性回归的学习与预测y=wx+b
regressor_poly2 = LinearRegression()
regressor_poly2.fit(x_train_poly2, y_train)
# 画出一次线性回归的拟合曲线
xx = np.linspace(0, 25, 100)
xx = xx.reshape(-1, 1)
xx_poly2 = poly2.fit_transform(xx)
yy_poly2 = regressor_poly2.predict(xx_poly2)

plt.scatter(x_train, y_train)
plt.plot(xx, yy_poly2, label="Degree2")
plt.show()

# 得出一条曲线

 

 

 

标签:plt,机器,多项式,拟合,xx,train,poly2,PolynomialFeatures
来源: https://www.cnblogs.com/xxdd123321/p/16541891.html