Text

TextClassificationDataModule

Bases: LightningDataModule

Lightning data module for HF text classification datasets.

This is based on this tutorial: https://lightning.ai/docs/pytorch/stable/notebooks/lightning_examples/text-transformers.html