class_docstr: A data record for storing attribution relevant information
code: def __init__(
      self,
      word_attributions,
      pred_prob,
      pred_class,
      true_class,
      attr_class,
      attr_score,
      raw_input_ids,
      convergence_score,
  ) -> None:
      self.word_attributions = word_attributions
      self.pred_prob = pred_prob
      self.pred_class = pred_class
      self.true_class = true_class
      self.attr_class = attr_class
      self.attr_score = attr_score
      self.raw_input_ids = raw_input_ids
      self.convergence_score = convergence_score
  
imports: import warnings
  from enum import Enum
  from typing import Any, Iterable, List, Optional, Tuple, Union
  import numpy
  from matplotlib import cm, colors, pyplot
  from matplotlib.collections import LineCollection
  from matplotlib.colors import LinearSegmentedColormap
  from matplotlib.figure import Figure
  from matplotlib.pyplot import axis, figure
  from mpl_toolkits.axes_grid1 import make_axes_locatable
  from numpy import ndarray
label_desc: Use image node_id 1 to create a new VisualizationDataRecord object  with example: obj = VisualizationDataRecord(word_attributions, pred_prob, pred_class, true_class, attr_class, attr_score, raw_input_ids, convergence_score)