Question about predictions on right-censored data

[tyler-keller]

Great library, thanks so much for it!

I'm currently using the DeepRecurrentSurvivalMachines model and I'm trying to predict the time-to-event values for my right-censored data. From testing, the values it's returning are much smaller than I'd expect.

Does .predict_mean() return these values as I'm expecting?

  def predict_mean(self, x, risk=1):
    r"""Returns the mean Time-to-Event \( t \)

    Parameters
    ----------
    x: np.ndarray
        A numpy array of the input features, \( x \).
    Returns:
      np.array: numpy array of the mean time to event.
    """

If not, could you let me know how I could achieve this functionality?