Closed hzhwcmhf closed 4 years ago
在forward函数中,data[gen_key]可以接受2d或3d数组 如果为2d,则每一个句子的reference只有1个,形状是[batch_size, ~gen_sentence_length] 若为3d,第2维则每个生成句子reference可以有多个,形状为 [batch_size, ~reference_num, ~gen_sentence_length] 注意成员变量reference_num会限制生成该3d数组第2维的大小:若为None,则大小任意;若为固定值,则第二维必须是referece_num那么大。
[batch_size, ~gen_sentence_length]
[batch_size, ~reference_num, ~gen_sentence_length]
同时,需要添加相应的测试:
提供参考:
目前我们还没有一个适用于这个的数据集dataloder。但一个理想的处理方案是, fields = OrderedDict([("references", SetenceCandidates)]) 这个field会提供多个reference,作为这个metric的输入
fields = OrderedDict([("references", SetenceCandidates)])
在forward函数中,data[gen_key]可以接受2d或3d数组 如果为2d,则每一个句子的reference只有1个,形状是
[batch_size, ~gen_sentence_length]
若为3d,第2维则每个生成句子reference可以有多个,形状为[batch_size, ~reference_num, ~gen_sentence_length]
注意成员变量reference_num会限制生成该3d数组第2维的大小:若为None,则大小任意;若为固定值,则第二维必须是referece_num那么大。同时,需要添加相应的测试:
提供参考:
目前我们还没有一个适用于这个的数据集dataloder。但一个理想的处理方案是,
fields = OrderedDict([("references", SetenceCandidates)])
这个field会提供多个reference,作为这个metric的输入