add new token classification model

ed302a73 · thomwolf · 89d47230 · ed302a73
Commit ed302a73 authored Nov 30, 2018 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

pytorch_pretrained_bert/__init__.py pytorch_pretrained_bert/__init__.py +2 -1

No files found.
--- a/pytorch_pretrained_bert/__init__.py
+++ b/pytorch_pretrained_bert/__init__.py
 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