Merge pull request #205 from lynschinzer/master

Fix divided by zero issue in case of words are not found in dict.
This commit is contained in:
Sun Junyi 2014-12-05 20:13:51 +08:00
commit b08c3f8ed7

View File

@ -21,7 +21,7 @@ class UndirectWeightedGraph:
ws = collections.defaultdict(float)
outSum = collections.defaultdict(float)
wsdef = 1.0 / len(self.graph)
wsdef = 1.0 / (len(self.graph) or 1.0)
for n, out in self.graph.items():
ws[n] = wsdef
outSum[n] = sum((e[2] for e in out), 0.0)