首頁  >  文章  >  後端開發  >  python如何實現線性迴歸

python如何實現線性迴歸

清浅
清浅原創
2019-04-08 13:15:4511487瀏覽

Python語言實作線性迴歸的步驟有:匯入所要用到的函式庫,讀取資料並進行預處理。分析資料以及建立線性迴歸模型,並進行模型訓練檢驗模型效果

透過使用python語言來實現線性迴歸是非常方便的,因為它提供了多個現成的函式庫,例如可以使用numpy.linalog .lstsq,pandas.ols以及

scipy.stats.linregress等,在本文中將使用sklearn函式庫的linear_model.LinearRegression,它支援任意維度,非常好用。

python如何實現線性迴歸

【推薦教學:Python教學

二維直線

例:線性方程式y=a∗x b;y=a∗x b 表示平面一直線

在下面的範例中,我們將建立線性迴歸模型,透過給出房屋的面積來預測房屋的價格

import pandas as pd
from io import StringIO
from sklearn import linear_model
import matplotlib.pyplot as plt
# 房屋面积与价格历史数据(csv文件)
csv_data = 'square_feet,price\n150,6450\n200,7450\n250,8450\n300,9450\n350,11450\n400,15450\n600,18450\n'
# 读入dataframe
df = pd.read_csv(StringIO(csv_data))
print(df)
# 建立线性回归模型
regr = linear_model.LinearRegression()
# 拟合
regr.fit(df['square_feet'].reshape(-1, 1), df['price']) # 注意此处.reshape(-1, 1),因为X是一维的!
# 不难得到直线的斜率、截距
a, b = regr.coef_, regr.intercept_
# 给出待预测面积
area = 238.5
# 方式1:根据直线方程计算的价格
print(a * area + b)
# 方式2:根据predict方法预测的价格
print(regr.predict(area))
# 画图
# 1.真实的点
plt.scatter(df['square_feet'], df['price'], color='blue')
# 2.拟合的直线
plt.plot(df['square_feet'], regr.predict(df['square_feet'].reshape(-1,1)), color='red', linewidth=4)
plt.show()

效果圖:

python如何實現線性迴歸

三維平面

線性方程式z= a∗x b∗y c;z=a∗x b∗y c 表示空間一平面

import numpy as np
from sklearn import linear_model
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
xx, yy = np.meshgrid(np.linspace(0,10,10), np.linspace(0,100,10))
zz = 1.0 * xx + 3.5 * yy + np.random.randint(0,100,(10,10))

# 构建成特征、值的形式
X, Z = np.column_stack((xx.flatten(),yy.flatten())), zz.flatten()

# 建立线性回归模型
regr = linear_model.LinearRegression()
# 拟合
regr.fit(X, Z)
# 不难得到平面的系数、截距
a, b = regr.coef_, regr.intercept_
# 给出待预测的一个特征
x = np.array([[5.8, 78.3]])
# 方式1:根据线性方程计算待预测的特征x对应的值z(注意:np.sum)
print(np.sum(a * x) + b)
# 方式2:根据predict方法预测的值z
print(regr.predict(x))
# 画图
fig = plt.figure()
ax = fig.gca(projection='3d')
# 1.画出真实的点
ax.scatter(xx, yy, zz)
# 2.画出拟合的平面
ax.plot_wireframe(xx, yy, regr.predict(X).reshape(10,10))
ax.plot_surface(xx, yy, regr.predict(X).reshape(10,10), alpha=0.3)
plt.show()

效果圖:

python如何實現線性迴歸

############################################################### #總結:以上就是這篇文章的全部內容了,希望對大家有幫助###### ###

以上是python如何實現線性迴歸的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn