Updated model, still getting weird results though

1 files changed, 5 insertions, 4 deletions

diff --git a/experiments/process.py b/experiments/process.py
index 1ec917e..b5b70ca 100644
--- a/experiments/process.py
+++ b/experiments/process.py
@@ -23,7 +23,7 @@ def build_network(filename):
                     victims[from_] = []
             else:
                 from_, to = int(float(row["from"])), int(float(row["to"]))
-                parent = (int(row["dist"]), int(row["t1"]))
+                parent = (int(row["dist"]), 3012 - int(row["t1"]))
                 if to not in victims:
                     non_victims[to] = []
                 non_victims[to].append(parent)
@@ -33,7 +33,7 @@ def build_network(filename):
     for victim in victims.keys():
         if not victims[victim]:
             del victims[victim]
-        root_victims[victim] = []
+            root_victims[victim] = []
     return root_victims, victims, non_victims
 
 
@@ -42,9 +42,10 @@ if __name__ == "__main__":
     #dump((root_victims, victims, non_victims), open("network.pickle", "w"))
     root_victims, victims, non_victims = load(open("network.pickle"))
 
-    alpha = np.arange(1000006., 1000007., 1.)
-    delta = np.arange(0.1, 1., 0.01)
+    alpha = np.arange(0.0000005, 0.00000051, 0.000001)
+    delta = np.arange(1., 1.000001, 0.001)
     with open("out.log", "a") as fh:
         for a, d in product(alpha, delta):
             beta, roots, ll = ml(root_victims, victims, non_victims, a, d)
             fh.write("\t".join(map(str, [a, d, beta, roots, ll])) + "\n")
+            fh.flush()