题解 | 使用正规方程的线性回归

使用正规方程的线性回归

https://www.nowcoder.com/practice/c497c2c49ea843898731bb14accc04ac

import numpy as np

def linear_regression_normal_equation(X, y):

X = np.array(X)

y = np.array(y)

results = np.linalg.lstsq(X, y, rcond=None)

theta = results[0]

# 将结果转换为扁平化列表并四舍五入

final_coeffs = [round(float(c), 4) for c in theta]

return final_coeffs

# 测试输入

X = [[1, 1], [1, 2], [1, 3]]

y = [2, 2, 3]

print(linear_regression_normal_equation(X, y))

# 实际输出应匹配预期输出: [1.3333, 0.5]

全部评论

相关推荐

04-03 09:32
已编辑
华南农业大学 golang
我的代码出BUG了:"晚点发个邮件调整一下时间",你收到新的邮件没,如果没有收到新的邮件,那就需要进入面试链接留痕,否则系统会判定你迟到
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务