How to access the true labels when using BatchLoader

[Mi-kaz]

Is there a simple way to access the true class labels so that they are in the same order as the predicted labels after calling predict() using a BatchLoader?

loader_te = BatchLoader(dataset = test, batch_size=batch_size, mask=True, shuffle=False)

predictions = modelGCN.predict(loader_te.load(), steps = loader_te.steps_per_epoch)