ENH: add model output to allow further prediction