added support for collecting and output predictions: PredictionOutput