class_docstr: Helper to visualize the clusters assignment of images
  
  This class is meant to be used in a notebook and will generate
  visualisations for the different clusters
class_docstr_tok: ["Helper","to","visualize","the","clusters","assignment","of","images","This","class","is","meant","to","be","used","in","a","notebook","and","will","generate","visualisations","for","the","different","clusters"]
code: def __init__(self, assignment: ClusterAssignment, split: str):
      self.config = assignment.config
      cluster_assignments = assignment.cluster_assignments[split]
      self.cluster_to_image = self._to_cluster_to_image_map(
          cluster_assignments
      )
      self.dataset = build_dataset(self.config, split)
      self.data_source = self.dataset.data_objs[0]
  
file_path: vissl/vissl/utils/cluster_utils.py
imports: import os
  from dataclasses import dataclass
  from typing import Dict, List
  import numpy
  import torch
  from iopath.common.file_io import g_pathmgr
  from vissl.config import AttrDict
  from vissl.data import build_dataset
label_desc: Use image node_id 1 to create a new ClusterVisualizer object  with example: obj = ClusterVisualizer(assignment, split)