import pkg_resources
import pprint
import evaluate
from ucdresolve import corpus
from ucdresolve import evaluation
from ucdresolve import ucdmodel

corpusName = "data/testData/testData10.txt"

def getBayesianCorpus(input):
	return corpus.BayesianCorpus.fromSource(input,
		ucdFactory=ucdmodel.VectorModelUCD.fromVizierLine)

trainingLines = pkg_resources.resource_stream('ucdresolve', corpusName
	).readlines()
testLines = pkg_resources.resource_stream('ucdresolve', 
	"data/testData/testData.txt").readlines()

for i in range(20):
	corpus.BayesianCorpus.expectationCorrection = 1+i/10.
	res = evaluation.EvaluationResult()
	evaluate.runOneEval(res, corpusName, getBayesianCorpus)
#	evaluation.evaluateModel(trainingLines, testLines, 
#		getBayesianCorpus, res, False)
	print str(corpus.BayesianCorpus.expectationCorrection)+" "+res.getResLine()