多項式回歸,回歸函式是回歸變數多項式的回歸。多項式回歸模型是線性回歸模型的一種,此時回歸函式關於回歸係數是線性的。由於任一函式都可以用多項式逼近,因此多項式回歸有著廣泛套用。
基本介紹
- 中文名:多項式回歸
- 外文名:Polynomial Regression
- 所屬學科:數理科學
- 類型:計算機科學
- 學科:跨學科
- 性質:回歸
- 概念:研究多項式的回歸分析
介紹
多項式回歸分析
套用
import matplotlib.pyplot as pltimport numpy as npfrom sklearn import linear_modelfrom sklearn.preprocessing import PolynomialFeatures
# 讀取數據集datasets_X = []datasets_Y = []fr = open('prices.txt','r')lines = fr.readlines()for line in lines: items = line.strip().split(',') datasets_X.append(int(items[0])) datasets_Y.append(int(items[1]))length = len(datasets_X)datasets_X = np.array(datasets_X).reshape([length,1])datasets_Y = np.array(datasets_Y)minX = min(datasets_X)maxX = max(datasets_X)X = np.arange(minX,maxX).reshape([-1,1])poly_reg = PolynomialFeatures(degree = 2)X_poly = poly_reg.fit_transform(datasets_X)lin_reg_2 = linear_model.LinearRegression()lin_reg_2.fit(X_poly, datasets_Y)
# 圖像中顯示plt.scatter(datasets_X, datasets_Y, color = 'red')plt.plot(X, lin_reg_2.predict(poly_reg.fit_transform(X)), color = 'blue')plt.xlabel('Area')plt.ylabel('Price')plt.show()