Entropy instance for LinearTransform

haskell · Jul 3, 2013 · 9a7364e · 9a7364e
1 parent 9eff104
commit 9a7364e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Statistics/Distribution/Transform.hs b/Statistics/Distribution/Transform.hs
@@ -72,6 +72,14 @@ instance (D.Variance d) => D.Variance (LinearTransform d) where
   variance (LinearTransform _ sc dist) = sc * sc * D.variance dist
   stdDev   (LinearTransform _ sc dist) = sc * D.stdDev dist
 
+instance (D.MaybeEntropy d, D.DiscreteDistr d) 
+         => D.MaybeEntropy (LinearTransform d) where
+  maybeEntropy (LinearTransform _ _ dist) = D.maybeEntropy dist
+
+instance (D.Entropy d, D.DiscreteDistr d) 
+         => D.Entropy (LinearTransform d) where
+  entropy (LinearTransform _ _ dist) = D.entropy dist
+
 instance D.ContGen d => D.ContGen (LinearTransform d) where
   genContVar (LinearTransform loc sc d) g = do
     x <- D.genContVar d g