R语言实现softmax

softmax <- function(a){
  max_a <- max(a)
  a <- a-max_a
  exp_a <- exp(a)
  sum_exp_a <- sum(exp_a)
  y <- exp_a/sum_exp_a
  return(y)
}
a <- c(1010,1000,990)
y <- softmax(a)
y
展开/折叠结果
[1] 9.999546e-01 4.539787e-05 2.061060e-09

评论

发表评论

了解 数据控|突破是我们的每一步 的更多信息

立即订阅以继续阅读并访问完整档案。

继续阅读