Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 783 Bytes

README.md

File metadata and controls

13 lines (9 loc) · 783 Bytes

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
  • the languages supported by XLNet-based classes
  • tokenization for text classification
  • utilities to load data, etc.
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.