
SimpleTransformers:适用于分类、命名实体识别、问答、语言建模、语言生成、T5、多模态及对话式AI的Transformer模型
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
SimpleTransformers是一个用户友好的库,支持多种预训练的Transformer模型用于文本分类、NER、QA等任务。同时它也适用于语言建模和生成、T5模型、多模态应用以及对话系统开发。
简单的变形金刚库基于HuggingFace的库构建。使用这个简化版Transformers库可以快速训练与评估Transformer模型。只需三行代码即可完成初始化、训练及评估任务。
该库支持的技术包括:
- 序列分类
- 代币分类(NER)
- 问题回答
- 语言模型微调
- 语言生成
- T5型号Seq2Seq任务
- 多模态分类
- 对话式AI
- 文本表示生成
安装方法如下:使用conda创建一个新的虚拟环境并安装所需的软件包。具体操作为:
```
conda create -n st python pandas tqdm
conda activate st
```
若要使用cuda,执行以下命令:
```
conda install pytorch>=1.6 cudatoolkit=11.0 -c pytorch
```
否则,请运行:
```
conda install pytorch cpuonly
```
全部评论 (0)
还没有任何评论哟~


