docs: def predict(self, data):
          """
          Predict new data for each group in the segmentation.
  
          Parameters
          ----------
          data : pandas.DataFrame
              Data to use for prediction. Must have a column with the
              same name as `segmentation_col`.
  
          Returns
          -------
          predicted : pandas.Series
              Predicted data in a pandas Series. Will have the index of `data`
              after applying filters.
  
          """
          with log_start_finish(
                  'predicting models in segmented model {}'.format(self.name),
                  logger):
              data = util.apply_filter_query(data, self.predict_filters)
              return self._group.predict(data)