import os
from PYEVALB import scorer
import sys


if len(sys.argv)==1:
    output = '/afs/inf.ed.ac.uk/group/project/prosody/prosody_nlp/code/self_attn_speech_parser/output/turn_part_fixed/'
    data = '/afs/inf.ed.ac.uk/group/project/prosody/prosody_nlp/data/input_features/turn_part_fixed/'
    gold = os.path.join(data,'turn_dev.trees')
    pred = os.path.join(output,'turn_sp_glove_72240_part_fixed_dev=86.34.pt_dev_predicted.txt')

else:
    gold = sys.argv[1]
    pred = sys.argv[2]

out = pred+'.eval'
scr = scorer.Scorer()
scr.evalb(gold,pred,out)

