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

Question

我有一个完整的使用此代码段的 bert 模型代码

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

鉴于我正在使用转换器工作，我如何导入 BertForSequenceClassification 而不是 bert 并处理相同的代码？谢谢

Answer 1

如果你使用的是huggingfacetransformers库，那么你可以按如下方式使用它：

from transformers import BertForSequenceClassification

BertForSequenceClassification 是 hugging-face 的 transformers 库中可用的 class。 BERT作者提供的BERT中没有

How do I import BERTforsequencclassification in same way importing bert?