线性回归系数

def solve():
    import pandas as pd
    from sklearn.linear_model import LinearRegression
    from sklearn.model_selection import train_test_split
    df = pd.read_csv("https://data.zuihe.com/regression.csv")
    X = df.drop("y", axis=1); y = df["y"]
    X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=42)
    lr = LinearRegression().fit(X_train, y_train)
    coefs = ",".join(f"{c:.4f}" for c in lr.coef_)
    print(f"{coefs};{lr.intercept_:.4f}")