docs: def predict(self, X):
          """Predict targets for a feature matrix.
  
          Args:
              X (np.array of float): feature matrix for prediction
  
          Returns:
              prediction (np.array)
          """
          logger.info('predicting ...')
          ps = self.predict_raw(X)
  
          return sigm(ps[:, 0])