1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import sys
from ml import ml
from ml2 import ml2
from ml3 import ml3
import numpy as np
from cPickle import load
from itertools import product
from math import exp
if __name__ == "__main__":
if len(sys.argv) < 2:
sys.exit("usage: {0} <file>".format(sys.argv[0]))
root_victims, victims, non_victims, age = load(open(sys.argv[1]))
# alphas = np.arange(5e-4, 3e-3, 1e-4) # parameter of the time component
alphas = np.logspace(-4,2,num=20)
# deltas = np.arange(0.03, 0.1, 0.005) # parameter of the structural component
deltas = np.logspace(-4,-1,num=10)
with open("out.log", "w") as fh:
for alpha, delta in product(alphas, deltas):
beta, roots, ll = ml3(root_victims, victims, non_victims, age, alpha, delta)
print "\t".join(map(str, [alpha, delta, beta, roots, ll]))
fh.write("\t".join(map(str, [alpha, delta, beta, roots, ll])) + "\n")
fh.flush()
# alpha = 0.0028
# delta = 0.06
# beta, roots, ll = ml2(root_victims, victims, non_victims, age, alpha, delta)
# print "\t".join(map(str, [1./alpha, delta, beta, roots, ll]))
|
