/transformers_tasks

NLP Algorithms with transformers lib. Supporting Text-Classification, Information-Extraction, Text-Matching, RLHF etc.

Primary LanguagePython

Transformers Tasks

基于 transformers 包实现的多种 NLP 任务。


已实现的任务模型：

1. 文本匹配（Text Matching）

模型 传送门
概览 [这里]
PointWise（单塔） [这里]
DSSM（双塔） [这里]
Sentence Bert（双塔） [这里]

2. 信息抽取（Information Extraction）

模型 传送门
UIE [这里]

3. Prompt任务（Prompt Tasks）

模型 传送门
PET [这里]
p-tuning [这里]

4. 文本分类（Text Classification）

模型 传送门
BERT-CLS [这里]

5. 强化学习 & 语言模型（Reinforcement Learning & Language Model）

模型 传送门
PPO-GPT2 [这里]