docs: def transform(self, X):
          """
          Parameters
          ----------
          X : array-like, shape [n x m]
              The mask in form of n x m array.
          """
          if self.mode_ == 'target':
              return np.apply_along_axis(self._target, 1, np.reshape(X, (X.shape[0], X.shape[1] * X.shape[2])))
  
          if self.mode_ == 'majority':
              return np.apply_along_axis(self._majority, 1, np.reshape(X, (X.shape[0], X.shape[1] * X.shape[2])))
  
          print('Invalid mode! Set mode to majority or target. Returning input.')
          return X