code: def _setup_datasets(
      dataset_name,
      root=".data",
      ngrams=1,
      vocab=None,
      include_unk=False,
  ):
      dataset_tar = download_from_url(URLS[dataset_name], root=root)
      extracted_files = extract_archive(dataset_tar)
  
      for fname in extracted_files:
          if fname.endswith("train.csv"):
              train_csv_path = fname
          if fname.endswith("test.csv"):
              test_csv_path = fname
  
      if vocab is None:
          logging.info(
              "Building Vocab based on {}".format(train_csv_path)
          )
          vocab = build_vocab_from_iterator(
              _csv_iterator(train_csv_path, ngrams)
          )
      else:
          if not isinstance(vocab, Vocab):
              raise TypeError("Passed vocabulary is not of type Vocab")
      logging.info("Vocab has {} entries".format(len(vocab)))
      logging.info("Creating training data")
      train_data, train_labels = _create_data_from_iterator(
          vocab,
          _csv_iterator(train_csv_path, ngrams, yield_cls=True),
          include_unk,
      )
      logging.info("Creating testing data")
      test_data, test_labels = _create_data_from_iterator(
          vocab,
          _csv_iterator(test_csv_path, ngrams, yield_cls=True),
          include_unk,
      )
      if len(train_labels ^ test_labels) > 0:
          raise ValueError("Training and test labels don't match")
      return (
          TextClassificationDataset(vocab, train_data, train_labels),
          TextClassificationDataset(vocab, test_data, test_labels),
      )
  
imports: import csv
  import io
  import logging
  import torch
  from torchtext.data.utils import get_tokenizer, ngrams_iterator
  from torchtext.utils import download_from_url, extract_archive
  from tqdm import tqdm
  from ..vocab import build_vocab_from_iterator, Vocab
label_desc: Use image node_id 3 for calling a global function with example usage: _setup_datasets(dataset_name, root, ngrams, vocab, include_unk) and returns: TextClassificationDataset,  TextClassificationDataset