code: def predict(self, X):
          
          proba = self.predict_proba(X)
          return self.classes_.take(np.argmax(proba, axis=1), axis=0)
docs: Predict classes on test vectors X.
  
          Parameters
          ----------
          X : array-like, shape = [n_samples, n_features]
              Input vectors, where n_samples is the number of samples
              and n_features is the number of features.
  
          Returns
          -------
          y : array, shape = [n_samples,]
              The predicted classes of the input samples.