【问题标题】:Using AllenNLP Interpret with a HuggingFace model将 AllenNLP 解释与 HuggingFace 模型一起使用
【发布时间】:2021-07-16 02:17:15
【问题描述】:

我想将 AllenNLP Interpret(代码 + 演示)与使用 HuggingFace(electra 基鉴别器)训练的 PyTorch 分类模型结合使用。然而,对我来说,如何转换我的模型并在本地 allen-nlp 演示服务器中使用它并不明显。

我应该如何进行?

提前致谢

【问题讨论】:

    标签: nlp huggingface-transformers allennlp


    【解决方案1】:

    如果您的任务是二进制分类,您可以查看https://github.com/allenai/allennlp-models/blob/main/training_config/classification/boolq_roberta.jsonnet 中的 BoolQ 示例。您可以更改该配置以使用不同的模型(例如 Electra)。

    我们还为解释功能发布了一些新文档:https://guide.allennlp.org/interpret

    为了给你一个更具体的答案,我需要了解更多细节,比如你要解决的任务是什么,你是如何训练原始模型的,等等。

    【讨论】:

    • 感谢您的快速回复!我将使用该配置文件进行测试,并查看解释新文档。我正在做一个简单的二进制分类,并且使用 HuggingFace Trainer 模块完成了培训。
    猜你喜欢
    • 2021-05-15
    • 2021-11-30
    • 2021-12-20
    • 2013-09-28
    • 2012-10-30
    • 1970-01-01
    • 2021-03-23
    • 2022-10-07
    • 2017-08-20
    相关资源
    最近更新 更多