Skip to main content

Transformers kit - Multi-task QA/Tagging/Multi-label Multi-Class Classification/Generation with BERT/ALBERT/T5/BERT

Project description




PyPI Download Build Last Commit CodeFactor Visitor

What is it

TFKit is a deep natural language process framework for classification/tagging/question answering/embedding study and language generation.
It leverages the use of transformers on many tasks with different models in this all-in-one framework.
All you need is a little change of config.

Task Supported

Classification Multi-class
Classification Multi-label
Question Answering Extractive - SQuAD like
Question Answering Multiple-choice
Tagging Sequence level
Tagging Sequence level with crf
Text Generation Seq2seq models - BART/T5/Bert2Bert...
Text Generation Causal LM models - GPT/GPT2...
Text Generation Once models
Text Generation Once models with ctc loss
Text Generation Onebyone models
Self-supervise Learning Mask LM

Getting Started

Learn more from the document.

Supplement

Contributing

Thanks for your interest.There are many ways to contribute to this project. Get started here.

License PyPI - License

Icons reference

Icons modify from Freepik from www.flaticon.com
Icons modify from Nikita Golubev from www.flaticon.com

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

tfkit-0.7.17.tar.gz (214.6 kB view hashes)

Uploaded Source

Built Distributions

tfkit-0.7.17-py3.7.egg (175.1 kB view hashes)

Uploaded Source

tfkit-0.7.17-py3-none-any.whl (78.0 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