Skip to main content

Transformer-based models implemented in tensorflow 2.x(Keras)

Project description

transformers-keras

Python package PyPI version Python

基于tf.keras的Transformers系列模型实现。

所有的Model都是keras模型,可以直接用于训练模型、评估模型或者导出模型用于部署。

在线文档:transformers-keras文档

本库功能预览:

  • 加载各种预训练模型的权重
  • 句向量的解决方案
  • 抽取式问答任务的解决方案
  • 序列分类任务的解决方案
  • 序列标注任务的解决方案

以上任务的数据处理、模型构建都完成了,可以一键开始训练。这里有两个例子:

更多使用方法和介绍,请查看在线文档:transformers-keras文档

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

transformers_keras-0.4.7.tar.gz (49.1 kB view hashes)

Uploaded Source

Built Distribution

transformers_keras-0.4.7-py3-none-any.whl (78.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page