code: def predict(self, X):
          
          if isinstance(X[0], list):
              return [self.estimator.tag(x) for x in X]
          return self.estimator.tag(X)
docs: Predict class labels for samples in X.
  
          Parameters
          ----------
          X : {array-like, sparse matrix}, shape = [n_samples, n_features]
              Samples.