import sys
from ml import ml
from ml2 import ml2
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(-5,-1,num=20)
    # # deltas = np.arange(0.03, 0.1, 0.005)  # parameter of the structural component
    # deltas = np.logspace(-5,-1,num=20)
    # with open("out.log", "w") as fh:
    #     for alpha, delta in product(alphas, deltas):
    #         beta, roots, ll = ml2(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.004
    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]))