下图是数据格式:前两列是属性值x1,x2,第三列是标签值y
代码如下
from numpy import genfromtxt from sklearn import linear_model # -----------------------读取数据--------------------------- dataPath = r"Delivery.csv" deliveryData = genfromtxt(dataPath,delimiter=',') print(deliveryData) # ------------------------准备数据--------------------------- x= deliveryData[:,:-1] y = deliveryData[:,-1] print(x) print(y) # --------------------------训练模型------------------------- lr = linear_model.LinearRegression() lr.fit(x, y) # -------对预测进行打分,用data_X预测的值和data_y进行对比------- print(lr.score(data_X, data_y)) # --------------------------x前面的系数----------------------- print(lr.coef_) # --------------------------截距------------------------------ print(lr.intercept_) # -------------------------------验证预测---------------------- xPredict = [102,6] yPredict = lr.predict(xPredict)print(yPredict)