Add experiment for number of evaluations.

1 files changed, 19 insertions, 0 deletions

diff --git a/sampling_alg_lac2020/figures/selection_count/calc_stats.py b/sampling_alg_lac2020/figures/selection_count/calc_stats.py
new file mode 100755
index 0000000..b192b72
--- /dev/null
+++ b/sampling_alg_lac2020/figures/selection_count/calc_stats.py
@@ -0,0 +1,19 @@
+#!/usr/bin/python3
+import sys
+
+#  if len(sys.argv) != 2:
+#          print("Wrong number of arguments.")
+#          print("USAGE: ./calc_stats.py <input_file>")
+#          quit()
+
+files = ["112_count.txt", "16_count.txt", "48_count.txt", "80_count.txt", "sweep_count.txt"]
+for input_file in files:
+	print(input_file)
+
+	f = open(input_file)
+	X = [int(i) for i in f.readlines()]
+
+	mean = float(sum(X))/len(X)
+	stddev = float(sum([x**2 for x in X]))/len(X) - mean**2
+
+	print("{0:.2f}".format(mean) + " " + "{0:.2f}".format(stddev))