Как реализовать Mixup в последовательности Keras?

Question

У меня мультиклассовая цель, и я пытаюсь реализовать метод смешивания в последовательности Keras.

Без этого я получаю более 70%, но ниже 5%, если добавляю код ниже.

Может ли кто-нибудь сообщить мне, что я делаю не так?

Спасибо!

def __getitem__(self, index):
    ...
    x_batch, y_batch = self.mixup(x_batch, y_batch)
    ...
    return x_batch, y_batch

def mixup(self, x, y):
    n = x.shape[0]
    l = np.random.beta(self.alpha, self.alpha, n)
    x_l = l.reshape(n, 1, 1, 1)
    y_l = l.reshape(n, 1)

    x1 = x
    x2 = np.flip(x, axis=0)
    x = x1 * x_l + x2 * (1 - x_l)

    y1 = y
    y2 = np.flip(y, axis=0)
    y = y1 * y_l + y2 * (1 - y_l)
    return x, y