模型

• 基于中文MengziT5的新闻评论生成模型
• 数据集来源于论文《Coherent Comment Generation for Chinese Articles with a Graph-to-Sequence Model》

生成评论

• 在线API只能生成一种评论，模型通过设置model.generate()参数是可以生成多种评论的

t5_tokenizer = T5Tokenizer.from_pretrained("Langboat/mengzi-t5-base")

model = T5ForConditionalGeneration.from_pretrained("wawaup/MengziT5-Comment")

def generate_comment(input_ids,cnt_num):
    outputs = model.generate(input_ids,
                            max_length=128,
                            do_sample=True,
                            temperature=0.9,
                            early_stopping=True,
                            repetition_penalty=10.0,
                            top_p=0.5,
                            num_return_sequences=cnt_num)
    print(outputs) 
    preds_cleaned = [t5_tokenizer.decode(ids, skip_special_tokens=True, 
                            clean_up_tokenization_spaces=True) for ids in outputs]
    print(preds_cleaned)
    return preds_cleaned