题解 | softmax激活函数实现

softmax激活函数实现

https://www.nowcoder.com/practice/438e5a90131b42a9b2caa0484cf6cf0b

import numpy as np
import math


def softmax(scores: list[float]) -> list[float]:
    exp_scores = np.exp(scores)
    probabilities = exp_scores/np.sum(exp_scores)
    return np.round(probabilities,4).tolist()

if __name__ == "__main__":
    scores = np.array(eval(input()))
    print(softmax(scores))

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务