code: def predict(self, X):
          
          return [self.classes[prediction.argmax()] for prediction in self.predict_proba(X)]
docs: Predict the class for X.
  
          The predicted class for each sample in X is returned.
  
          Parameters
          ----------
          X : List of ndarrays, one for each training example.
              Each training example's shape is (string1_len,
              string2_len, n_features), where string1_len and
              string2_len are the length of the two training strings and
              n_features the number of features.
  
          Returns
          -------
          y : iterable of shape = [n_samples]
              The predicted classes.