Finish classification explanations

A couple things I missed in 7f68dc6fc6
This commit is contained in:
Samuel Sloniker 2023-04-16 15:48:19 -07:00
parent d8f3d2e701
commit 2c3fc77ba6
Signed by: kj7rrv
GPG Key ID: 1BB4029E66285A62
2 changed files with 4 additions and 1 deletions

View File

@ -95,7 +95,10 @@ class Model:
else:
explanation[word] = {
"weight": weight,
"probabilities": word_probs,
"probabilities": {
name: word_probs[index]
for index, name in enumerate(self.names)
},
"count": 1,
}

Binary file not shown.