class_docstr: Dataset class for quick embedding/label retrieval.
  Labels should be in the index space.
  
  If the labels provided are not in torch form, will convert them.
code: def __init__(self, encodings, labels):
      self.encodings = encodings
      self.labels = labels
  
file_path: lightwood/lightwood/encoder/text/helpers/pretrained_helpers.py
imports: import torch
  from transformers import AdamW
label_desc: Use image node_id 1 to create a new TextEmbed object from inherited base classes: torch.utils.data with example: obj = TextEmbed(encodings, labels)