def __init__(self, input_nodes, output_nodes, sess, tokenizer, settings, minus):
Base.__init__(
self,
input_nodes=input_nodes,
output_nodes=output_nodes,
sess=sess,
tokenizer=tokenizer,
label=None,
)
self._tag2idx = settings
self._idx2tag = {int(v): k for k, v in self._tag2idx.items()}
self._minus = minus
def __init__(
self,
input_nodes,
output_nodes,
sess,
tokenizer,
label=['not similar', 'similar'],
):
Base.__init__(
self,
input_nodes=input_nodes,
output_nodes=output_nodes,
sess=sess,
tokenizer=tokenizer,
label=label,
)
def __init__(
self, input_nodes, output_nodes, sess, tokenizer, settings, tok=None
):
Base.__init__(
self,
input_nodes=input_nodes,
output_nodes=output_nodes,
sess=sess,
tokenizer=tokenizer,
label=None,
)
self._settings = settings
self._tok = tok
self._settings['idx2tag'] = {
int(k): v for k, v in self._settings['idx2tag'].items()
}
def __init__(
self,
input_nodes,
output_nodes,
sess,
tokenizer,
module,
label=['negative', 'positive'],
):
Base.__init__(
self,
input_nodes=input_nodes,
output_nodes=output_nodes,
sess=sess,
tokenizer=tokenizer,
label=label,
module=module,
)