【发布时间】:2020-11-02 20:46:55
【问题描述】:
我使用此代码段对 bert 模型进行了完整的代码工作
import bert #BertForSequenceClassification
from bert import run_classifier
from bert import optimization
from bert import tokenization
鉴于我正在使用转换器,我如何导入 BertForSequenceClassification 而不是 bert 并处理相同的代码? 谢谢
【问题讨论】:
-
如果你使用的是 huggingface 转换器库,那么你可以按如下方式使用它:
from transformers import BertForSequenceClassification -
是的,但是usnig run_clasifier,优化和tokenization出现错误,因为它们与BertForSequenceClassification不兼容
-
您使用的是哪个
run_classifier.py? -
来自 BERT 的 TF Hub 可加载模块
-
BertForSequenceClassification是transformers拥抱脸库中的一个类。它在 BERT 作者提供的 BERT 中不可用。
标签: import bert-language-model