WIP: Faster predictions if n_outputs==1