通过分析匹萨的直径与价格的数据的线性关系,来预测任意直径匹萨的价格
# 直径(英寸) X = [[6], [8], [10], [12], [14], [18], [20]] # 价格(美元) Y = [[7], [9], [12], [15], [18], [20], [24]]
通过分析匹萨的直径与价格的数据的线性关系,来预测任意直径匹萨的价格
# 直径(英寸) X = [[6], [8], [10], [12], [14], [18], [20]] # 价格(美元) Y = [[7], [9], [12], [15], [18], [20], [24]]
登录后即可发布作业,立即登录
全部作业 30
以下是使用 Python 语言,基于线性回归来实现根据上述数据预测匹萨价格的示例:
python
import numpy as np
from sklearn.linear_model import LinearRegression
X = np.array([[6], [8], [10], [12], [14], [18], [20]])
Y = np.array([7, 9, 12, 15, 18, 20, 24])
model = LinearRegression()
model.fit(X, Y)
new_diameter = np.array([[16]])
predicted_price = model.predict(new_diameter)
print(f"直径为16英寸的匹萨预测价格为:{predicted_price[0]:.2f}美元")
代码解释:
导入必要库:
numpy 用于数值计算,处理数组等数据结构。
LinearRegression 来自 sklearn.linear_model,是用于实现线性回归的类。
准备数据:将直径数据 X 转换为二维数组形式(符合 sklearn 中线性回归模型对自变量的要求),价格数据 Y 为一维数组。
创建并训练模型:
创建 LinearRegression 模型对象 model。
使用 fit 方法将直径数据 X 和价格数据 Y 输入模型进行训练,让模型学习两者之间的线性关系。
进行预测:构造一个新的直径数据(这里是 16 英寸)的二维数组 new_diameter,使用训练好的模型的 predict 方法预测其对应的价格,并打印输出预测结果。