Add mean_field_from_activation for SoftmaxUnits.

hayesgb · Mar 25, 2013 · 905969d · 905969d
1 parent 518c1bb
commit 905969d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/morb/units.py b/morb/units.py
@@ -134,6 +134,9 @@ def probabilities(self, vmap):
     def sample_from_activation(self, vmap):
         p = self.probabilities_from_activation(vmap)
         return samplers.multinomial(p)
+
+    def mean_field_from_activation(self, vmap):
+        return self.probabilities_from_activation(vmap)
 
 
 class SoftmaxWithZeroUnits(Units):