'''
Created on Jul 4, 2017

@author: David
'''
import numpy as np
import pickle
import sys
def cmap(pred):
    if pred == "A1":
        return 0
    if pred == "A2":
        return 1
    if pred == "B1":
        return 2
    if pred == "B2":
        return 3
    if pred == "C1":
        return 4
    return -1

def f1(p,r):
    return (2*p*r)/(p+r)

def mean(l):
    return float(sum(l))/max(len(l),1)

def recalculate(pname,data_file,verbose=True):
    clf = pickle.load(open(pname, "rb"))
    o = open("svalex-r-nh.csv","w")
    with open(data_file, "r") as f:
        for l in f:
            if not l.strip():
                continue
            values,target = l.rstrip().rsplit(",", 1)
            values_np = np.array(values.split(","), dtype='float64').reshape(1,-1)
            pred = clf.predict(values_np)[0]
            st = values + "," + pred
            o.write(st)
            o.write("\n")
    return None

if __name__ == "__main__":
    recalculate(sys.argv[1], sys.argv[2])