https://lacucinadiadine.blogspot.com/2024/03/how-to-use-softmax-output-in-python-for.html