Python LinearCRF.LinearCRF Examples

Example #1

File: neuralcrf.py Project: ZPeng-Research/pytorch_lstmcrf

 def __init__(self, config, print_info: bool = True):
     super(NNCRF, self).__init__()
     self.embedder = WordEmbedder(config, print_info=print_info)
     self.encoder = BiLSTMEncoder(config,
                                  self.embedder.get_output_dim(),
                                  print_info=print_info)
     self.inferencer = LinearCRF(config, print_info=print_info)

Example #2

 def __init__(self, config, print_info: bool = True):
     super(TransformersCRF, self).__init__()
     self.device = config.device
     self.embedder = TransformersEmbedder(config, print_info=print_info)
     if config.hidden_dim > 0:
         self.encoder = BiLSTMEncoder(config, self.embedder.get_output_dim(), print_info=print_info)
     else:
         self.encoder = LinearEncoder(config, self.embedder.get_output_dim(), print_info=print_info)
     self.inferencer = LinearCRF(config, print_info=print_info)