当前位置:   article > 正文

【NLP】提升文本生成多样性的实用方法

【NLP】提升文本生成多样性的实用方法

比如用T5模型,训练数据是inputText-outputText格式,预测时do_sample=False

# 预测代码

from transformers import TFAutoModelForSeq2SeqLM
from transformers import AutoTokenizer


checkpoint_local = "./path/"
tokenizer = AutoTokenizer.from_pretrained(checkpoint_local)
model = TFAutoModelForSeq2SeqLM.from_pretrained(checkpoint_local, 
                                        
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/871117
推荐阅读
相关标签
  

闽ICP备14008679号