今天學習了簡單的線性迴歸，用波士頓房價的例子測試了一下

import pandas as pd
import numpy as np
from sklearn import datasets
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

#把資料轉化成pandas的形式，在列尾加上房價PRICE
boston_dataset=datasets.load_boston()
data=pd.DataFrame(boston_dataset.data)
data.columns=boston_dataset.feature_names
data['PRICE']=boston_dataset.target

#取出房間數和房價並轉化成矩陣形式
x=data.loc[:,'RM'].as_matrix(columns=None)
y=data.loc[:,'PRICE'].as_matrix(columns=None)

#進行矩陣的轉置
x=np.array([x]).T
y=np.array([y]).T

#訓練線性模型
l=LinearRegression()
l.fit(x,y)

#畫圖顯示
plt.scatter(x,y,s=10,alpha=0.5,c='green')
plt.plot(x,l.predict(x),c='blue',linewidth='1')
plt.xlabel("X")
plt.ylabel("Y")
plt.show()
 

結果如下