Python实现Softmax与交叉熵函数及损失函数对比分析

周末不加班
周末不加班
发布时间:2026-10-14| 6人浏览过
原创

本文提供Python实现的Softmax函数与交叉熵损失函数代码示例,并深入解析分类任务中为何优先使用交叉熵而非均方误差(MSE)。通过对比分析指出,交叉熵仅关注正确分类结果,而MSE会受错误分类影响,导致训练效率差异。

Python实现Softmax与交叉熵函数

以下是使用Python编写的Softmax函数代码示例:


import numpy as np
 
# Write a function that takes as input a list of numbers, and returns
# the list of values given by the softmax function.
def softmax(L):
 pass
 expL = np.exp(L)
 sumExpL = sum(expL)
 result = []
 for i in expL:
  result.append(i*1.0/sumExpL)
 return result

接下来展示Python中交叉熵公式的具体实现代码:


import numpy as np
 
def cross_entropy(Y, P):
 Y = np.float_(Y)
 P = np.float_(P)
 return -np.sum(Y * np.log(P) + (1 - Y) * np.log(1 - P))

分类任务中损失函数的选择:交叉熵与MSE对比

补充说明:在分类任务中,通常不采用均方误差(MSE),而是选择交叉熵作为损失函数,其原因如下。

MSE(均方误差)会对所有输出结果赋予同等重视,而交叉熵则主要关注正确分类的预测结果。

举例说明:假设在一个三分类模型中,模型的预测输出为(a, b, c),而真实标签为(1, 0, 0)。此时,MSE与交叉熵对应的损失函数计算值分别如下:

MSE的计算结果:

交叉熵的计算结果:

从上述公式可以看出,交叉熵损失函数仅与正确分类的预测结果相关,而MSE损失函数则同时受到错误分类结果的影响。MSE在促使正确分类概率增大的同时,还会试图让错误分类的概率趋于平均,但在实际分类问题中,这种对错误分类的“平均化”调整并无必要。

然而,对于回归问题而言,这种对所有输出结果的均衡考量则显得至关重要。因此,在回归任务中使用交叉熵并不合适。

以上关于Python编写Softmax函数与交叉熵函数的实例分享完毕,希望能为读者提供参考。

相关标签:

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。