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:
|
else:
|
||||||
explanation[word] = {
|
explanation[word] = {
|
||||||
"weight": weight,
|
"weight": weight,
|
||||||
"probabilities": word_probs,
|
"probabilities": {
|
||||||
|
name: word_probs[index]
|
||||||
|
for index, name in enumerate(self.names)
|
||||||
|
},
|
||||||
"count": 1,
|
"count": 1,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Binary file not shown.
Loading…
Reference in New Issue
Block a user