将 AllenNLP 解释与 HuggingFace 模型一起使用

Posted 2023-03-29

【中文标题】将 AllenNLP 解释与 HuggingFace 模型一起使用

【英文标题】：Using AllenNLP Interpret with a HuggingFace model

【发布时间】：2021-07-16 02:17:15

【问题描述】：

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

我应该如何进行？

提前致谢

【参考方案1】：

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

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

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

【讨论】：

感谢您的快速回复！我将使用该配置文件进行测试，并查看解释新文档。我正在做一个简单的二进制分类，并且使用 HuggingFace Trainer 模块完成了培训。