- 例子
一家快递公司送货:X1: 运输里程 X2: 运输次数 Y:总运输时间
目的,求出b0, b1,…. bp:
y_hat=b0+b1x1+b2x2+ … +bpxp
- Python代码:
from numpy import genfromtxt
import numpy as np
from sklearn import linear_model
deliveryData=genfromtxt("Delivery.csv",delimiter=',')
print('data:',deliveryData)
X=deliveryData[:,:-1]
Y=deliveryData[:,-1]
print('X:',X)
print('Y:',Y)
regr=linear_model.LinearRegression()
regr.fit(X, Y);
print('coeffients',regr.coef_)
print('intercept:',regr.intercept_)
x=[102,6]
y=regr.predict([x])
print('predicted y:',y)