机器学习(四)-回归模型评估指标

文章目录

    • [1. 哪个模型更好?](#1. 哪个模型更好?)
    • [2. 线性回归评估指标](#2. 线性回归评估指标)
    • [3. python 实现线性模型评估指标](#3. python 实现线性模型评估指标)

1. 哪个模型更好?

我们之前已经对房价预测的问题构建了线性模型,并对测试集进行了预测。

如图所示,横坐标是地区人口,纵坐标是房价,红色的点是实际样本分布。

使用不同的算法或者策略构建了两个线性模型,如图,分布是绿色直线和黄色直线所示。

那么如何量化我们构建的线性回归模型性能呢?如何比较这2个模型哪个更好呢?本节我们将给大家介绍一下线性回归常用的评估指标。

2. 线性回归评估指标

线性回归常用的评估指标有4个,下面详细介绍!

Var 为方差。

R2是我们常用的线性回归评估指标,主要的原因之一就是我们通过R2能更好的理解模型的性能好坏。

R^2有以下几个特点:

3. python 实现线性模型评估指标

python 复制代码
# 线性模型评估指标
# 均方误差:MSE
from sklearn.metrics import mean_squared_error
MSE = mean_squared_error(y_test, y_pred)
print("MSE = {}".format(MSE))

# 均方根误差:RMSE
RMSE = np.sqrt(MSE)
print("RMSE = {}".format(RMSE))

# 平均绝对值误差:MAE
from sklearn.metrics import mean_absolute_error
MAE = mean_absolute_error(y_test, y_pred)
print("MAE = {}".format(MAE))

# R^2: R-Squared
from sklearn.metrics import r2_score
R2 = r2_score(y_test, y_pred)
print("R2 = {}".format(R2))

输出结果:

python 复制代码
MSE = 0.17473032823222068
RMSE = 0.4180075695872273
MAE = 0.3406988798247714
R2 = 0.9838332745214976
相关推荐
开发者工具分享4 分钟前
缺乏项目进度可视化手段,如何提升展示效果
人工智能
慧星云13 分钟前
ComfyUI工作流 :一键二次元角色转真人
人工智能·云计算·aigc
聚客AI17 分钟前
⚡ 突破LLM三大局限:LangChain架构核心解析与最佳实践
人工智能·langchain·llm
Sui_Network21 分钟前
tBTC 现已上线 Sui,带来 5 亿美元的比特币流动性
人工智能·物联网·web3·区块链·量子计算
春哥的研究所1 小时前
AI人工智能名片小程序源码系统,名片小程序+分销商城+AI客服,包含完整搭建教程
人工智能·微信小程序·小程序
ahead~1 小时前
【大模型入门】访问GPT_API实战案例
人工智能·python·gpt·大语言模型llm
喜欢吃豆1 小时前
深入企业内部的MCP知识(三):FastMCP工具转换(Tool Transformation)全解析:从适配到增强的工具进化指南
java·前端·人工智能·大模型·github·mcp
pany1 小时前
写代码的节奏,正在被 AI 改写
前端·人工智能·aigc
我爱一条柴ya1 小时前
【AI大模型】神经网络反向传播:核心原理与完整实现
人工智能·深度学习·神经网络·ai·ai编程
万米商云1 小时前
企业物资集采平台解决方案:跨地域、多仓库、百部门——大型企业如何用一套系统管好百万级物资?
大数据·运维·人工智能