XLNet-based Classes
This folder contains utility functions and classes based on the implementation of Transformers.
Summary
The following table summarizes each Python script.
Script | Description |
---|---|
common.py | This script includes <ul><li>the languages supported by XLNet-based classes</li><li> tokenization for text classification</li> <li>utilities to load data, etc.</li></ul> |
sequence_classification.py | An implementation of sequence classification based on fine-turning XLNet. It is commonly used for text classification. The module includes logging functionality using MLFlow. |
utils.py | This script includes a function to visualize a confusion matrix. |