Lijiadong / TorchBlocks

A PyTorch-based toolkit for natural language processing

TorchBlocks

A PyTorch-based toolkit for natural language processing

Requirements

• transformers>=4.1.1
• torch>=1.6.0

TorchBlocks requires Python 3.6+. We recommend installing TorchBlocks in a Linux or OSX environment.

Installation

Recommended (because of active development):

git clone https://github.com/lonePatient/TorchBlocks.git
cd TorchBlocks
python setup.py install

Tutorials

• Tutorial 1 (text classification): task_text_classification_cola.py
• Tutorial 2 (siamese similarity): task_siamese_similarity_afqmc.py
• Tutorial 3 (sequence labeling): task_sequence_labeling_ner_crf.py
• Tutorial 4 (sentence similarity): task_sentence_similarity_lcqmc.py
• Tutorial 5 (triple similarity): task_triple_similarity_epidemic.py
• Example scripts for each task: TorchBlocks/examples/