class_docstr: Token Classification Model with BERT, applicable for tasks such as Named Entity Recognition
class_docstr_tok: ["Token","Classification","Model","with","BERT",",","applicable","for","tasks","such","as","Named","Entity","Recognition"]
docstr: Update data directory and get data stats with Data Descriptor
  Weights are later used to setup loss
  
  Args:
      data_dir: path to data directory
docstr_tok: ["Update","data","directory","and","get","data","stats","with","Data","Descriptor","Weights","are","later","used","to","setup","loss","Args",":","data_dir",":","path","to","data","directory"]
code: def update_data_dir(self, data_dir: str) -> None:
      """
      Update data directory and get data stats with Data Descriptor
      Weights are later used to setup loss
  
      Args:
          data_dir: path to data directory
      """
      self._cfg.dataset.data_dir = data_dir
      logging.info(f"Setting model.dataset.data_dir to {data_dir}.")
  
imports: import os
  from typing import List, Optional, Union
  import torch
  from omegaconf import DictConfig, OmegaConf
  from pytorch_lightning import Trainer
  from torch.utils.data import DataLoader
  from nemo.collections.common.losses import CrossEntropyLoss
  from nemo.collections.nlp.data.data_utils.data_preprocessing import get_labels_to_labels_id_mapping
  from nemo.collections.nlp.data.token_classification.token_classification_dataset import BertTokenClassificationDataset, BertTokenClassificationInferDataset
  from nemo.collections.nlp.data.token_classification.token_classification_utils import get_label_ids
  from nemo.collections.nlp.metrics.classification_report import ClassificationReport
  from nemo.collections.nlp.models.nlp_model import NLPModel
  from nemo.collections.nlp.modules.common import TokenClassifier
  from nemo.collections.nlp.parts.utils_funcs import get_classification_report, plot_confusion_matrix, tensor2list
  from nemo.core.classes.common import PretrainedModelInfo, typecheck
  from nemo.utils import logging
label_desc: Use image node_id 2 for calling the TokenClassificationModel obj's underlying member method code with example usage: obj.update_data_dir(data_dir) without return types