docs: def predict(self, X):
          """ Predict class of X.
  
          The predicted class for each sample in X is returned.
  
          Parameters
          ----------
  
          X : array-like of shape = [n_samples, n_features]
              The input samples.
  
          Returns
          -------
          y : array of shape = [n_samples]
              The predicted classes,
          """
          #TODO: Check consistency of X
          if self.pruned:
              tree_ = self.tree_.tree_pruned
          else:
              tree_ = self.tree_.tree
  
          return self._classify(X, tree_, proba=False)