import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']
import numpy as np
wen_du = np.asarray([[25], [28], [31], [33], [35]])
bing_qi_lin = np.asarray([[110], [115], [155], [160], [180]])
yuce_wendu = np.asarray([[30],[28],[55]])
# x = [1,2,3,4,5]
# y = [7,10,13,16,19]
# 1。导入人工智能工具包 linear 线性(直线) 模型
from sklearn import linear_model
# 2。创建一个模型对象 类class Regression 回归,推算
moXing = linear_model.LinearRegression()
# 3.训练模型 fit训练
moXing.fit(wen_du,bing_qi_lin)
# 4.预测数据 predict 预测
yucezhi = moXing.predict(yuce_wendu)
print(yucezhi)
# 5.给模型打分 分数score
fenshu = moXing.score(wen_du,bing_qi_lin)
print(fenshu)
plt.plot(wen_du,bing_qi_lin,"bo")
plt.plot(yuce_wendu,yucezhi,"ro")
# plt.plot(x,y,"bo")
plt.xlabel("温度")
plt.ylabel("冰淇淋")
plt.show()
boren - 用机器学习的线性回归模型预测冰淇淋销量
最新推荐文章于 2021-12-22 22:37:52 发布