Actually take the difference

pymc-devs · Jul 22, 2017 · ae7d265 · ae7d265
1 parent 8d06ab0
commit ae7d265
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc3/stats.py b/pymc3/stats.py
@@ -932,4 +932,4 @@ def bfmi(trace):
     """
     energy = trace['energy']
 
-    return np.square(energy).mean() / np.var(energy)
+    return np.square(np.diff(energy)).mean() / np.var(energy)