change ln to log2 to have entropy in bits

2017-12-03 11:30:09 +01:00 · 2017-12-03 11:30:09 +01:00 · 2f1937d25b
parent a720a5ce5c
commit 2f1937d25b
1 changed files with 1 additions and 1 deletions
--- a/andrei.py
+++ b/andrei.py
@ -100,7 +100,7 @@ def build_model(trans):
        model[state] = Node(
            choices=tuple(x[0] for x in ord),
            cumdist=tuple(accumulate([x[1] for x in ord])),
-            entropy=-sum(f/tot * log(f/tot) for f in succs.values()))
+            entropy=-sum(f/tot * log(f/tot, 2) for f in succs.values()))
    return model