diff options
| author | Soumith Chintala <soumith@gmail.com> | 2015-08-03 21:46:06 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2015-08-03 21:46:06 +0300 |
| commit | e3452817c8b4dd9a5d0836c2bb1b4e46641c6f60 (patch) | |
| tree | 3c249cf8ac26a5034427b07f038d41a4843310fb | |
| parent | e5da24d3b011eb8ee6a8f003a6d844f75947ec36 (diff) | |
| parent | 5d971696f4653c80220c2a084e229e228d6b3c11 (diff) | |
Merge pull request #63 from jonathantompson/confusion_matrix_fix
Change ConfusionMatrix accumulator from Float to Long.
| -rw-r--r-- | ConfusionMatrix.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ConfusionMatrix.lua b/ConfusionMatrix.lua index 3b5fa9c..b4fa38d 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 |