Python GNMTGlobalScorer.GNMTGlobalScorer示例

示例#1

文件： predictor.py 项目： AILAN110/QA-Abstract-And-Reasoning

def build_predictor(args, tokenizer, symbols, model, logger=None):
    scorer = GNMTGlobalScorer(args.alpha, length_penalty="wu")

    translator = Translator(
        args, model, tokenizer, symbols, global_scorer=scorer, logger=logger
    )
    return translator

示例#2

文件： test_pretrain.py 项目： XinnuoXu/AggGen

def build_predictor(args, model, logger=None):
    scorer = GNMTGlobalScorer(args.alpha, length_penalty='wu')
    translator = Translator(args, model, global_scorer=scorer, logger=logger)
    return translator