Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'scripts')
-rwxr-xr-xscripts/average.py7
1 files changed, 2 insertions, 5 deletions
diff --git a/scripts/average.py b/scripts/average.py
index dacd614b..4d5c0a2f 100755
--- a/scripts/average.py
+++ b/scripts/average.py
@@ -12,12 +12,9 @@ for filename in sys.argv[1:-1]:
m = np.load(mfile)
for k in m:
if k not in average:
- average[k] = 1 / m[k]
+ average[k] = m[k] / n
elif average[k].shape == m[k].shape:
- average[k] += 1 / m[k]
-
-for k in average:
- average[k] = n / average[k]
+ average[k] += m[k] / n
print "Saving to", sys.argv[-1]
np.savez(sys.argv[-1], **average)