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(1e-3, 1e-2, 8e-4)  # parameter of the time component
    alphas = np.logspace(-4,-.8,num=20)
    # deltas = np.arange(0.001, 0.3, 0.008)  # parameter of the structural component
    deltas = np.logspace(-2,-.0001,num=20)
    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]))