code: def __init__(
      self, figure_path, data, title, x_label=None, y_label=None
  ):
      if len(data) < 3:
          fig_size = (5, 5)
      else:
          fig_size = (10, 10)
      super(HeatMap, self).__init__(
          file_path=figure_path,
          data=data,
          title=title,
          figure_size=fig_size,
          x_label=x_label,
          y_label=y_label,
      )
  
imports: import matplotlib.pyplot
  import numpy
  import pandas
  import seaborn
  import xai.constants
  from wordcloud import WordCloud
  import shap
  from xai.graphs.basic_graph import Graph
  from typing import List
  from collections import Counter
  import operator
  from xai.data.exceptions import NoItemsError
label_desc: Use image node_id 1 to create a new HeatMap object from inherited base classes: Graph with example: obj = HeatMap(figure_path, data, title, x_label, y_label)