Entropy quantities for non-proj trees

yzhangcs · yzhangcs · commit 50bd7884230c · 2021-12-26T19:43:11.000+08:00
diff --git a/supar/structs/tree.py b/supar/structs/tree.py
@@ -69,7 +69,7 @@ def sample(self):
 
     @lazy_property
     def entropy(self):
-        raise NotImplementedError
+        return self.log_partition - (self.marginals * self.scores).sum((-1, -2))
 
     def cross_entropy(self, other):
         raise NotImplementedError
