diff options
author | nicholas-leonard <nick@nikopia.org> | 2014-06-03 20:00:37 +0400 |
---|---|---|
committer | nicholas-leonard <nick@nikopia.org> | 2014-06-03 20:00:37 +0400 |
commit | f2b1a60fad4ab4cb8c51287743837142e21aabd4 (patch) | |
tree | 0196ab342c5a69bb47e66d4e87d8e002b2fcbc5d /SoftMaxTree.lua | |
parent | ccd03d9820a78bfff07068997bab77c293d263de (diff) |
fixed SoftMaxTree.__init bug
Diffstat (limited to 'SoftMaxTree.lua')
-rw-r--r-- | SoftMaxTree.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/SoftMaxTree.lua b/SoftMaxTree.lua index 09284a5..ce5e2bf 100644 --- a/SoftMaxTree.lua +++ b/SoftMaxTree.lua @@ -113,7 +113,7 @@ function SoftMaxTree:__init(inputSize, hierarchy, rootId, verbose) local nChildren = self.parentChildren[nodeId][2] treeSize, pathSize = getSize(parentId) treeSizes[parentId] = treeSize + nChildren - pathSizes[parentId] = size + 1 + pathSizes[parentId] = pathSize + 1 end return treeSize, pathSize end |