This folder contains utility functions and classes based on the implementation of Transformers.
The following table summarizes each Python script.
Script | Description |
---|---|
common.py | This script includes
|
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. |