用python来实现机器学习（一）：线性回归（linear regression）

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了用python来实现机器学习（一）：线性回归（linear regression），小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1946字，纯文字阅读大概需要3分钟。

内容图文

需要下载一个data：auto-mpg.data

第一步：显示数据集图

import pandas as pd
import matplotlib.pyplot as plt
columns = ["mpg","cylinders","displacement","horsepower","weight","acceleration","model year","origin","car name"]
cars = pd.read_table("E:/3_python_code/python_ml/data/auto-mpg.data",delim_whitespace=True,names=columns)
print (cars.head(5))
#加了两个子图
fig = plt.figure()
ax1 = fig.add_subplot(2,1,1)
ax2 = fig.add_subplot(2,1,2)
cars.plot("weight","mpg",kind="scatter",ax=ax1)
cars.plot("acceleration","mpg",kind="scatter",ax=ax2)
plt.show()

第二步：训练并预测

import pandas as pd
import matplotlib.pyplot as plt
columns = ["mpg","cylinders","displacement","horsepower","weight","acceleration","model year","origin","car name"]
cars = pd.read_table("E:/3_python_code/python_ml/data/auto-mpg.data",delim_whitespace=True,names=columns)
# print (cars.head(5))
# #加了两个子图
fig = plt.figure()
# ax1 = fig.add_subplot(2,1,1)
# ax2 = fig.add_subplot(2,1,2)
# cars.plot("weight","mpg",kind="scatter",ax=ax1)
# cars.plot("acceleration","mpg",kind="scatter",ax=ax2)
# plt.show()

import sklearn
from sklearn.linear_model import  LinearRegression
lr = LinearRegression()
#训练
lr.fit(cars[["weight"]],cars["mpg"])
#预测
predictions = lr.predict(cars[["weight"]])
from sklearn.metrics import mean_squared_error
#均方误差
mse = mean_squared_error(cars["mpg"],predictions)
print(mse)
print(predictions[0:5])
print(cars["mpg"][0:5])
plt.scatter(cars["weight"],cars["mpg"],c="red")
plt.scatter(cars["weight"],predictions,c="blue")
plt.show()

输出结果 显示均方误差18.78，如下

18.7809397346
[ 19.41852276  17.96764345  19.94053224  19.96356207  19.84073631]
0    18.0
1    15.0
2    18.0
3    16.0
4    17.0
Name: mpg, dtype: float64
Process finished with exit code 0

![在这里插入图片描述](https://www.icode9.com/i/ll/?i=20181127115834245.png?,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyMTY2Nzc5,size_16,color_FFFFFF,t_70)

内容总结

以上是互联网集市为您收集整理的用python来实现机器学习（一）：线性回归（linear regression）全部内容，希望文章能够帮你解决用python来实现机器学习（一）：线性回归（linear regression）所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端