AssertionError: A prediction has class=10, but the dataset only has 10 classes and predicted class id should be in [0, 9].

AssertionError: A prediction has class=10, but the dataset only has 10 classes and predicted class id should be in [0, 9]. #1241