Okay. so I’m assuming that:

For all n:  if i <= 5, output[n, i] corresponds with a probability for label i.

output[n, 6] corresponds to the label 7

output[n, 7] corresponds to the label 8

output[n, 8] corresponds to the label 9

If this is true you can map your targets to the thing they correspond with:
def map_labels(label):
    if label > 6:
        return label - 1
    return label