geometric average

1 files changed, 5 insertions, 5 deletions

diff --git a/scripts/average.py b/scripts/average.py
index ecfcd097..3f3eb091 100755
--- a/scripts/average.py
+++ b/scripts/average.py
@@ -5,16 +5,16 @@ import numpy as np;
 
 average = dict()
 
+n = len(argv[1:-1])
 for filename in sys.argv[1:-1]:
+    print "Loading", filename 
     with open(filename, "rb") as mfile:
         m = np.load(mfile)
         for k in m:
             if k not in average:
-                average[k] = m[k]
+                average[k] = m[k] ** 1.0/n
             elif average[k].shape == m[k].shape:
-                average[k] += m[k]
-
-for k in average:
-    average[k] /= len(sys.argv[1:-1])
+                average[k] *= m[k] ** 1.0/n
 
+print "Saving to", sys.argv[-1]
 np.savez(sys.argv[-1], **average)