python-3.x - 使用 Spacy Textcat 自定义损失函数

Question

我已经四处寻找了一段时间了。我想知道是否可以修改/自定义 spaCy textcategorizer 的损失函数。

我的意思是，当你想要提取模型(例如 BERT)并想要在损失函数中添加回归组件来优化(考虑每个类的概率而不是仅标签)时，我不明白在哪里我应该寻找。我尝试探索一些spaCy code但只有一个函数可以得到损失。

如果有人知道在哪里寻找可视化损失函数并更改它(例如通过编写子类)，那就太好了!

谢谢

阿尔诺

Answer 1

SpaCy 最终构建于 thinc 之上因此，如果您想要进行自定义工作，您应该使用 Thinc，而不是 SpaCy。 SpaCy 通常允许您使用原始 Thinc 模型初始化管道。

特别是因为 SpaCy 的理念是提供一种运行良好的实现，而不一定是一个 super 可定制的框架。