def METHOD_NAME(self, x: np.ndarray, batch_size: int = 128, **kwargs):
    """
    Perform prediction of the neural network for samples `x`.
    :param x: Samples of shape (nb_samples, nb_features) or (nb_samples, nb_pixels_1, nb_pixels_2,
              nb_channels) or (nb_samples, nb_channels, nb_pixels_1, nb_pixels_2).
    :param batch_size: Batch size.
    :return: Predictions.
    :rtype: Format as expected by the `model`
    """
    return NeuralNetworkMixin.METHOD_NAME(self, x, batch_size=batch_size, **kwargs)