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

import numpy as np

def linear_regression_normal_equation(X:list[list[float]], y:list[float]) -> list[float]:
    X = np.array(X)
    y = np.array(y).reshape(-1, 1)
    X_transpose = X.T

    theta = np.linalg.inv(X_transpose.dot(X)).dot(X_transpose).dot(y)

    theta = np.round(theta, 4).flatten().tolist()

    return theta

if  __name__ == '__main__':
    import ast
    X = ast.literal_eval(input())
    y = ast.literal_eval(input())
    res = linear_regression_normal_equation(X,y)
    print(res)