Exemplos de BertWrapper.BertWrapper em Python

Exemplo n.º 1

 def build_model(self):
     """Construct the model."""
     num_classes = len(self.class_list)
     return BertWrapper(BertModel.from_pretrained(self.pretrained_path),
                        num_classes)

Exemplo n.º 2

Arquivo: bert_classifier.py Projeto: yonghangzhou/KBRD

 def build_model(self):
     num_classes = len(self.class_list)
     self.model = BertWrapper(
         BertModel.from_pretrained(self.opt['bert_id']), num_classes)