C++实现softmax函数的面试经验
更新时间：2022年05月18日 12:57:17 作者：concyclics
这篇文章主要为大家介绍了C++实现softmax函数的面试经验,有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
背景
今天面试字节算法岗时被问到的问题，让我用C++实现一个softmax函数。softmax是逻辑回归在多分类问题上的推广。大概的公式如下：
即判断该变量在总体变量中的占比。
第一次实现
实现
我们用vector来封装输入和输出，简单的按公式复现。
vector<double> softmax(vector<double> input)
{
double total=0;
for(auto x:input)
{
total+=exp(x);
}
vector<double> result;
for(auto x:input)
{
result.push_back(exp(x)/total);
}
return result;
}
测试
test 1
- 测试用例1: {1, 2, 3, 4, 5}
- 测试输出1: {0.0116562, 0.0316849, 0.0861285, 0.234122, 0.636409}
经过简单测试是正常的。
test 2
但是这时面试官提出了一个问题，即如果有较大输入变量时会怎么样？
- 测试用例2: {1, 2, 3, 4, 5, 1000}
- 测试输出2: {0, 0, 0, 0, 0, nan}
由于 e^1000已经溢出了双精度浮点(double)所能表示的范围，所以变成了NaN(not a number)。
第二次实现(改进)
改进原理
我们注意观察softmax的公式:
如果我们给上下同时乘以一个很小的数，最后答案的值是不变的。
那我们可以给每一个输入 x i 都减去一个值 a ，防止爆精度。
大致表示如下:
实现
vector<double> softmax(vector<double> input)
{
double total=0;
double MAX=input[0];
for(auto x:input)
{
MAX=max(x,MAX);
}
for(auto x:input)
{
total+=exp(x-MAX);
}
vector<double> result;
for(auto x:input)
{
result.push_back(exp(x-MAX)/total);
}
return result;
}
测试
test 1
- 测试用例1: {1, 2, 3, 4, 5, 1000}
- 测试输出1: {0, 0, 0, 0, 0, 1}
test 2
- 测试用例1: {0, 19260817, 19260817}
- 测试输出1: {0, 0.5, 0.5}
我们发现结果正常了。
完整代码
#include <iostream>
#include <vector>
#include <math.h>
using namespace std;
vector<double> softmax(vector<double> input)
{
double total=0;
double MAX=input[0];
for(auto x:input)
{
MAX=max(x,MAX);
}
for(auto x:input)
{
total+=exp(x-MAX);
}
vector<double> result;
for(auto x:input)
{
result.push_back(exp(x-MAX)/total);
}
return result;
}
int main(int argc, char *argv[])
{
int n;
cin>>n;
vector<double> input;
while(n--)
{
double x;
cin>>x;
input.push_back(x);
}
for(auto y:softmax(input))
{
cout<<y<<' ';
}
}
以上就是C++实现softmax函数的面试经验的详细内容，更多关于C++ softmax函数的资料请关注脚本之家其它相关文章！
相关文章
这个是大一时期写的。大四的时候整理了一下（本人C语言学的也不太好）。肯定很多不足和存在漏洞的地方、仅供借鉴、仅供借鉴，代码中有大量注释，新手看起来也没有困难2021-11-11
今天小编就为大家分享一篇关于解析C语言结构体及位段，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2018-12-12
结构体是一些值的集合，这些值称为成员变量，结构体的每个成员可以是不同类型的变量。本文将通过示例为大家详细讲讲C语言中结构体的使用，需要的可以参考一下2022-07-07
对于C/C++初学者来说,可能会经常遇到如计算阶乘等问题,下面这篇文章主要给大家介绍了关于C语言求阶乘之和的三种实现方法,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-07-07
这篇文章主要介绍了算法之排列算法与组合算法详解,本文以字典序法、递归法为例讲解了排列算法、全组合算法等,需要的朋友可以参考下2014-08-08
今天小编就为大家分享一篇关于关于C++友元类的实现讲解，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2018-12-12
VisualStudio2019配置OpenCV4.5.0的方法示例
这篇文章主要介绍了VisualStudio2019配置OpenCV4.5.0的方法示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-03-03
Qt提供了两种定时器,一种是QObject类的定时器,另一种是QTimer类的定时器,这篇文章主要给大家介绍了关于Qt定时器类QTimer使用与注意事项的相关资料,需要的朋友可以参考下2023-10-10
这篇文章主要介绍了C语言编写获取Linux本地目录及本机信息的小程序实例,小程序能够根据参数输出目录的结构以及获取主机用户的信息,需要的朋友可以参考下2016-04-04
我们在学习编程的过程中，虽然有些语法很简单，但是我们还是要做多题。不做题是发现不了问题的，发现问题我们就可以“对症下药”，进行查漏补缺了。刷题可以先从简单题开始刷，熟练之后再做一些可以提升自己能力的题2022-07-07
最新评论