How to sort multiclass model predictions?

[minhtran2801]

Hi all,

I have a multi-class model and I use 4 paragraphs in a document for predictions. How would I sort their similarity in order? Thank you.

[luizniero]

Hellow. You need to change the predict() method to use a softmax function (predict is using a argmax). I made this change and pull request at: #1360