【问题标题】:How do I import BERTforsequencclassification in same way importing bert?如何以与导入 bert 相同的方式导入 BERTforsequencclassification?
【发布时间】: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 可加载模块
  • BertForSequenceClassificationtransformers 拥抱脸库中的一个类。它在 BERT 作者提供的 BERT 中不可用。

标签: import bert-language-model


【解决方案1】:

如果你使用的是 huggingface transformers 库,那么你可以按如下方式使用它:

from transformers import BertForSequenceClassification

BertForSequenceClassification 是变形金刚拥抱脸库中的一个类。 BERT 的作者提供的 BERT 中没有此功能

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2014-01-09
    • 2021-02-16
    • 2017-12-14
    • 2021-10-28
    • 2022-10-13
    • 2017-11-16
    • 2014-09-01
    • 1970-01-01
    相关资源
    最近更新 更多