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