Finish classification explanations
A couple things I missed in 7f68dc6fc6
This commit is contained in:
parent
d8f3d2e701
commit
2c3fc77ba6
|
@ -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.
Loading…
Reference in New Issue
Block a user