add new token classification model

This commit is contained in:
thomwolf 2018-11-30 22:55:03 +01:00
parent 89d47230d7
commit ed302a73f4
1 changed files with 2 additions and 1 deletions

View File

@ -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