python机器学习之线性回归（linear regression）

部分数据展示

TIPS

代码实现

import matplotlib.pyplot as plt
import pandas as pd
from IPython.core.display import display
from envs import liner_regression
from sklearn.model_selection import train_test_split
from sklearn import linear_model

path='Advertising.csv'
data = pd.read_csv(path)
x = data[['TV','radio','newspaper']]
y = data['sales']

# 数据可视化
plt.figure(figsize=(9,12))
plt.subplot(311)
plt.plot(data['TV'],y,'ro')
plt.title('TV')
plt.grid()
plt.subplot(312)
plt.plot(data['radio'],y,'g^')
plt.title('radio')
plt.grid()
plt.subplot(313)
plt.plot(data['newspaper'],y,'b*')
plt.title('newspaper')
plt.grid()
plt.tight_layout()
plt.show()

# 默认%75训练，%25测试
x_train, x_test, y_train, y_test = train_test_split(x, y, random_state=1)
model=linear_model.LinearRegression()
model.fit(x_train,y_train)

# 输出intercept(截距)和coef(系数1*3矩阵)
display(model.intercept_)
display(model.coef_)

# 预测
y_pred = model.predict(x_test)
plt.figure(figsize=(9,12))
plt.plot(range(len(y_pred)), y_pred, 'b', label='predict')
plt.plot(range(len(y_pred)), y_test, 'r', label='test')
plt.legend(loc="upper right")
plt.xlabel("number of sales")
plt.ylabel("value of sales")
plt.show()

data visualization

intercept&coefficient

Prediction

声明：部分数据/图片来自网络，如有侵权联系博主删除