def get_activations(X, batch_iterator, func):
    activations = []

    for Xb, yb in batch_iterator(X):
        activations.append(NeuralNet.apply_batch_func(func, Xb))

    return activations
def get_activations(X, batch_iterator, func):
    activations = []

    for Xb, yb in batch_iterator(X):
        activations.append(NeuralNet.apply_batch_func(func, Xb))

    return activations