allennlp - 在配置中传递损失函数和指标

Question

在官方示例中，指标和损失函数都是硬编码的。我想知道我们是否可以在配置 jsonnet 中传递这些，这样我就可以在具有不同指标的不同数据集中重用我的模型。

标签： allennlp

Answer 1

我知道我以前见过这个问题。从GitHub复制和粘贴：

Metric是可注册的，因此您可以轻松地向您的模型添加参数 type List[Metric]，然后在 Jsonnet 中指定指标。您必须确保这些指标采用完全相同的输入。

对于损失，这有点困难。您将创建自己的Registrable基类，然后以这种方式实现您想要使用的损失。您可以使用Metric该类作为如何执行此操作的示例。这将是一些打字工作，但并不困难。