diff options
| author | Soumith Chintala <soumith@gmail.com> | 2015-10-04 06:22:13 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2015-10-04 06:22:13 +0300 |
| commit | c8a42f4fbe874ca6e896ce31b145807a1a34aa8c (patch) | |
| tree | 6d0d62f0df0ab0f94413adb124eecc440e999a9c | |
| parent | 7dfaad9019f7a1808b2f1516ebde1dd775a8c3c3 (diff) | |
proper fix for out of range (on large number of classes) in confusion matrixconffix
| -rw-r--r-- | ConfusionMatrix.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ConfusionMatrix.lua b/ConfusionMatrix.lua index 1467649..bf5747a 100644 --- a/ConfusionMatrix.lua +++ b/ConfusionMatrix.lua @@ -17,7 +17,7 @@ function ConfusionMatrix:__init(nclasses, classes) classes = nclasses nclasses = #classes end - self.mat = torch.FloatTensor(nclasses,nclasses):zero() + self.mat = torch.LongTensor(nclasses,nclasses):zero() self.valids = torch.FloatTensor(nclasses):zero() self.unionvalids = torch.FloatTensor(nclasses):zero() self.nclasses = nclasses @@ -226,7 +226,7 @@ end function ConfusionMatrix:render(sortmode, display, block, legendwidth) -- args - local confusion = self.mat + local confusion = self.mat:double() local classes = self.classes local sortmode = sortmode or 'score' -- 'score' or 'occurrence' local block = block or 25 |