AssertionError: 预测的类别为 10, 但数据集只包含 10 个类别, 预测的类别 ID 应在 [0, 9] 内。