Python-線性迴歸

 使用2018年台南每個月平均溫度及每月主要觀光遊憩據點遊客人數，來預估當溫度20度時，來觀光人數約多少。

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26

import numpy as np from sklearn.linear_model import LinearRegression #載入CSV資料。 dataset = np.genfromtxt('D:\\Train\\travel.csv', delimiter=',', dtype=None, names=('Temperature','People'), skip_header=1, encoding='UTF-8') #print(dataset.dtype.names) #設定溫度及人數。 Temperature = dataset['Temperature'] People_Number = dataset['People'] #設定線性迴歸 Line_Re = LinearRegression() #進行線性迴歸，算出截距及斜率 Line_Re.fit(np.reshape(Temperature, (len(Temperature), 1)), np.reshape(People_Number, (len(People_Number), 1))) #截距 print('截距:',Line_Re.intercept_) #鈄率 print('斜率:',Line_Re.coef_) # 輸入溫度進行預測。 Number = input('Please enter temperature:') New_Temperature=np.reshape(int(Number),(-1,1)) #預測結果。 predicted = Line_Re.predict(New_Temperature) print('當溫度：',New_Temperature[0][0],'度,預則人數',int(predicted))

結果：