def __init__(self, knowledgebase):
if _sanity_check:
ExprGenerator.setup()
State.ExtraInfoGen = ExprGenerator
Perceptron.model = Model()
Perceptron.indepKB = IndepKnowledgeBase()
Perceptron.KB = knowledgebase
knowledgebase.postedit_indepkb(Perceptron.indepKB)
Perceptron.c = 0
Perceptron.iter = FLAGS.iter
Perceptron.beamsize = FLAGS.beam
Perceptron.parser = Parser(Perceptron.indepKB, Perceptron.KB, Perceptron.model, State)
Perceptron.ncpus = FLAGS.ncpus
Perceptron.ontheflyfd = FLAGS.ontheflyfd
Perceptron.single_gold = FLAGS.singlegold
Perceptron.output_prefix = FLAGS.outputprefix
Perceptron.fdbeamsize = FLAGS.fdbeam
if Perceptron.ncpus > 0:
Perceptron.shared_memo_size = int(1024 * 1024 * 1024) # 1G shared memory
Perceptron.shared_memo = mmap.mmap(-1, Perceptron.shared_memo_size,
mmap.MAP_SHARED, mmap.PROT_READ | mmap.PROT_WRITE)
Perceptron.ref_beams = {}
if FLAGS.ref:
print >> LOGS, "loading refs",
hgs = pickle.load(open(FLAGS.ref))
self.load_hgs(hgs)
if FLAGS.extraref:
print >> LOGS, "loading extra refs",
hgs = pickle.load(open(FLAGS.extraref))
self.load_hgs(hgs)
评论列表
文章目录