如何以与导入bert相同的方式导入BERTforsequencclassification？

Question

我使用此代码段对bert模型进行了完整的代码研究

import bert #BertForSequenceClassification
from bert import run_classifier
from bert import optimization
from bert import tokenization

考虑到我使用的是转换器，如何导入BertForSequenceClassification而不是bert并使用相同的代码？ 谢谢

Answer 1

如果您使用的是拥抱面transformers库，则可以按以下方式使用它：

from transformers import BertForSequenceClassification

BertForSequenceClassification是可在拥抱面的变形金刚库中使用的类。 BERT的作者提供的BERT中没有此功能