提交 04d27ee9 编写于 作者: H Hongyi 提交者: Jiangtao Hu

Prediction: minor fix softmax

上级 d152c686
......@@ -45,10 +45,12 @@ std::vector<double> Softmax(const std::vector<double>& value) {
std::vector<double> result;
double sum = 0.0;
for (std::size_t i = 0; i < value.size(); ++i) {
sum += std::exp(value[i]);
double exp_value = std::exp(value[i]);
sum += exp_value;
result.push_back(exp_value);
}
for (std::size_t i = 0; i < value.size(); ++i) {
result.push_back(std::exp(value[i]) / sum);
result[i] = result[i] / sum;
}
return result;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册