You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何用DataFrame以表格形式展示线性回归模型的MAE、MSE和R2?

用DataFrame展示线性回归模型指标的方法

首先纠正你代码里的两个小问题:

  • MAE和MSE的计算函数搞反了:mean_absolute_error对应MAE,mean_squared_error对应MSE
  • r2_score的参数顺序要注意:真实值y_test在前,预测值predictions在后,结果才准确

下面是生成规整DataFrame表格的完整方案:

完整代码示例

import pandas as pd
from sklearn.metrics import mean_squared_error, mean_absolute_error, r2_score

# 修正指标计算逻辑
LinearRegression_MAE = mean_absolute_error(y_test, predictions)
LinearRegression_MSE = mean_squared_error(y_test, predictions)
LinearRegression_R2 = r2_score(y_test, predictions)

# 构造指标数据字典
metrics_data = {
    "指标名称": ["MAE", "MSE", "R²"],
    "线性回归模型": [LinearRegression_MAE, LinearRegression_MSE, LinearRegression_R2]
}

# 转换为DataFrame表格
Report = pd.DataFrame(metrics_data)

# 打印查看结果
print(Report)

运行后会输出类似这样的表格:

指标名称  线性回归模型
0   MAE    2.156
1   MSE   12.345
2    R²    0.892

可选:索引式表格形式

如果想要更紧凑的展示方式,可以把指标作为DataFrame的索引:

metrics_data = {
    "线性回归模型": {
        "MAE": LinearRegression_MAE,
        "MSE": LinearRegression_MSE,
        "R²": LinearRegression_R2
    }
}
Report = pd.DataFrame(metrics_data)

输出效果:

线性回归模型
MAE      2.156
MSE     12.345
R²       0.892

生成的Report是标准的DataFrame对象,你可以直接用于查看、导出为CSV/Excel文件,或者结合其他工具做可视化分析。

内容的提问来源于stack exchange,提问作者Ramsey A.

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.13 01:41:18