我已经训练了一个多标签(2),多类(3)模型与keras和model.predict(test_data)给3(num_samples,1)ndarray ...到目前为止很好,但我怎么能检查哪个预测数组是针对哪个类的:我的模型架构是:

Input - Embedding - GRU1 - BatchNorm - Dropout - GRU2 - Dense(1) for Class1 - Dense(1) for Class2 - Dense(1) for Class3

Keras docs仅提及predict()返回'Numpy array(s) of predictions'但没有关于维护订单的信息 . 谢谢,