Fix divided by zero issue in case of words are not found in dict.

This commit is contained in:
Lin 2014-12-05 17:13:12 +08:00
parent 293dbbc390
commit fea3aec6bd

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)