devarajphukan / multi-class-text-classifier-hackerrank Goto Github PK

hi,
this code works great for me,thank you.
But I have a question: How can I specify the probability of belonging a sentence to each class ?
for example I test this sentence: "this is a book" and the classifier says it belongs to class=1.
but I want to know something like this"
it belongs to class=1 with 76.44
it belongs to class=2 with 6.44
it belongs to class=3 with 7.47
.
.
.

could you please help me.