diff options
Diffstat (limited to 'ConfusionMatrix.lua')
-rw-r--r-- | ConfusionMatrix.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ConfusionMatrix.lua b/ConfusionMatrix.lua index 3026968..cb5cea6 100644 --- a/ConfusionMatrix.lua +++ b/ConfusionMatrix.lua @@ -43,6 +43,7 @@ function ConfusionMatrix:add(prediction, target) local _,target = target_1d:max(1) self.mat[target[1]][prediction[1]] = self.mat[target[1]][prediction[1]] + 1 end + collectgarbage() -- the code above is a bit naive, and ends up allocating a lot end function ConfusionMatrix:zero() |