最小二乘法一次拟合

数值分析第三章16题

 答案

import numpy as np
from sklearn.linear_model import LinearRegression

# 创建输入数据
X = np.array([[0], [0.9], [1.9], [3.0], [3.9],[5.0]])
y = np.array([0, 10, 30, 50, 80,110])

# 创建线性回归模型
model = LinearRegression()

# 训练模型
model.fit(X, y)

# 打印回归方程系数和截距
print("斜率: ", model.coef_[0])
print("截距: ", model.intercept_)