diff options
| author | Soumith Chintala <soumith@gmail.com> | 2015-09-16 17:25:26 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2015-09-16 17:25:26 +0300 |
| commit | 93e75e0939092df1feead61f2c873a73275e8178 (patch) | |
| tree | d12b52fdb7669af2f47bfa15cd473a8c1ba64c14 | |
| parent | a1f517075df0d9607bb6472eeb3d623680e2fcbf (diff) | |
Revert "Change ConfusionMatrix accumulator from Float to Long."revert-63-confusion_matrix_fix
| -rw-r--r-- | ConfusionMatrix.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ConfusionMatrix.lua b/ConfusionMatrix.lua index b4fa38d..3b5fa9c 100644 --- a/ConfusionMatrix.lua +++ b/ConfusionMatrix.lua @@ -17,7 +17,7 @@ function ConfusionMatrix:__init(nclasses, classes) classes = nclasses nclasses = #classes end - self.mat = torch.LongTensor(nclasses,nclasses):zero() + self.mat = torch.FloatTensor(nclasses,nclasses):zero() self.valids = torch.FloatTensor(nclasses):zero() self.unionvalids = torch.FloatTensor(nclasses):zero() self.nclasses = nclasses |