利用sklearn linear_model下的線性模型驗證一下特徵縮放先後結果是否一致app
驗證因變量y的縮放spa
import numpy as np from sklearn import linear_model x=[[1],[2],[3],[4],[5]] y=[100,200,300,400,500] y2=[] avg=np.mean(y) for i in range(len(x)): y2.append(y[i]/avg) clf=linear_model.LinearRegression() X=[[2]] clf.fit(x,y) start=clf.predict(X) clf.fit(x,y2) end=clf.predict(X) print(start) print(end*avg)
結果:code
[200.]
[200.]blog
驗證自變量x的縮放(結上面的代碼)it
y=[[1],[2],[3],[4],[5]] x=[[100],[200],[300],[400],[500]] x2=[[0],[0],[0],[0],[0]] avg=np.mean(x) for i in range(len(x)): x2[i][0]=x[i][0]/avg clf=linear_model.LinearRegression() X=[[100]] clf.fit(x,y) start=clf.predict(X) clf.fit(x2,y) end=clf.predict(X/avg) print(start) print(end)
結果:io
[[1.]]
[[1.]]
>>> class
縮放先後的結果是一致的~~~import