train_classes = np.unique(y_train) test_classes = np.unique(y_test) if not np.array_equal(train_classes, test_classes): print('Train and test classes are not the same!') print(f'Train classes: {train_classes}') print(f'Test classes: {test_classes}') exit(1)

这段代码是用来检查训练集和测试集的类别是否相同。首先，使用NumPy库中的unique函数获取训练集和测试集中的所有不同类别。然后，使用NumPy库中的array_equal函数比较两个数组是否相等。如果两个数组不相等，则打印出训练集和测试集的类别，并退出程序。这段代码的目的是确保在模型训练和测试过程中使用的类别相同，以避免错误的预测结果。