fix it so model weights for each prediction are output in NA case