data.csv

    1. from sklearn.model_selection import train_test_split
    2. import numpy as np
    3. from sklearn import datasets,linear_model
    4. import pandas as pd
    6. data =  pd.read_csv("data.csv“）
    7. y = data.iloc[:,2]
    8. x = data.iloc[:, 3:5]
    9. # 训练数据
    10. x_train, x_test, y_train, y_test = train_test_split(x, y, test_size = 0.25, random_state = 1)
    11. regr = linear_model.LinearRegression()
    12. regr.fit(x_train,y_train)
    14. print('coefficients(b1,b2...):',regr.coef_)
    15. print('intercept(b0):',regr.intercept_)
    16. # 预测数据
    17. y_pred = regr.predict(x_test)
    18. # 查看模型得分
    19. regr.score(x_test, y_test)