https://mingredbestcheapprice.blogspot.com/2023/10/how-to-use-softmax-output-in-python-for.html