问题描述
我使用seq2seq模型,它可以在每个时期计算BLEU得分(NMT得分)。但是,我无法将BLEU分数设置为验证指标,因此它不能及早停止训练。我阅读了源代码,但是没有任何暗示可以向验证指标中添加哪种字符串(“ +损失”除外)。请救救我,谢谢!
解决方法
默认的validation_metric
实际上是"-loss"
,而不是"+loss"
。 “-”表示该指标应最小化,而不是最大化。
因此,要改用BLEU分数,请将validation_metric
设置为"+BLEU"
。
通常,您可以使用模型的.get_metric()
方法返回的任何度量。您用于validation_metric
的度量标准名称仅需与.get_metric()
返回的字典中的相应键相匹配。
在您的情况下,大概是模型的.get_metric()
方法返回如下内容:{"BLEU": ...}
,这就是为什么validation_metric
应该设置为"+BLEU"
的原因。