docs: def draw(self):
          """
          Renders the class balance chart on the specified axes from support.
          """
          # Number of colors is either number of classes or 2
          colors = resolve_colors(len(self.support_))
  
          if self._mode == BALANCE:
              self.ax.bar(
                  np.arange(len(self.support_)), self.support_,
                  color=colors, align='center', width=0.5
              )
  
          # Compare mode
          else:
              bar_width = 0.35
              labels = ["train", "test"]
  
              for idx, support in enumerate(self.support_):
                  index = np.arange(len(self.classes_))
                  if idx > 0:
                      index = index + bar_width
  
                  self.ax.bar(
                      index, support, bar_width,
                      color=colors[idx], label=labels[idx]
                  )
  
          return self.ax