def get_position_scorer(name, lang, sent_pos, doc_size):
graph = Scorer(name)
graph.add_input(lang).add_input(sent_pos).add_input(doc_size)
graph.add_hidden(50, HIDDEN_ACT).add_hidden(50,
HIDDEN_ACT) # 2 hidden layers
graph.add_output(1, tf.nn.sigmoid)
return graph.get_output()
def create_scorer(name, inputs, nbr_outs):
graph = Scorer(name)
for input in inputs:
graph.add_input(input)
graph.add_hidden(
50, HIDDEN_ACT) #.add_hidden(10, HIDDEN_ACT) # 2 hidden layers
graph.add_output(nbr_outs, tf.nn.sigmoid)
return graph.get_output()
def get_sentence_scorer(name, lang, tfreq, sim, size, pos):
graph = Scorer(name)
graph.add_input(lang)
graph.add_input(tfreq)
graph.add_input(sim)
graph.add_input(size)
graph.add_input(pos)
graph.add_hidden(50, HIDDEN_ACT).add_hidden(20,
HIDDEN_ACT) # 2 hidden layers
graph.add_output(1, tf.nn.sigmoid)
return graph.get_output()
def __init__(self):
Scorer.__init__(self)
self.exact = False
def __init__(self):
Scorer.__init__(self)
self.match_type = "overlap"