本次实验使用的是flan-t5-base模型,不支持中文。
from transformers import T5Tokenizer, T5ForConditionalGeneration
tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base", device_map="auto")
1.翻译
2.情感识别
3. 提问
4. 单个实体提取
时间实体
地点实体
扫描二维码关注公众号,回复:
16022559 查看本文章
深度实体
5. 单个问题生成
6. 生成摘要
7. 产生相似问
8. 数学计算
9. 逻辑关系