Improving prediction output generation