From ed302a73f4cc365db4cd29d26ca722d605bc85a1 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Fri, 30 Nov 2018 22:55:03 +0100 Subject: [PATCH] add new token classification model --- pytorch_pretrained_bert/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch_pretrained_bert/__init__.py b/pytorch_pretrained_bert/__init__.py index 7850fa5555..fc9b15a12d 100644 --- a/pytorch_pretrained_bert/__init__.py +++ b/pytorch_pretrained_bert/__init__.py @@ -1,6 +1,7 @@ from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer from .modeling import (BertConfig, BertModel, BertForPreTraining, BertForMaskedLM, BertForNextSentencePrediction, - BertForSequenceClassification, BertForQuestionAnswering) + BertForSequenceClassification, BertForTokenClassification, + BertForQuestionAnswering) from .optimization import BertAdam from .file_utils import PYTORCH_PRETRAINED_BERT_CACHE