Function all_activity

bindsnet/evaluation/evaluation.py:109–150

Classify data with the label with highest average spiking activity over all neurons. :param spikes: Binary tensor of shape ``(n_samples, time, n_neurons)`` of a layer's spiking activity. :param assignments: A vector of shape ``(n_neurons,)`` of neuron label assignments. :pa

(
    spikes: torch.Tensor, assignments: torch.Tensor, n_labels: int
)

def all_activity(
    spikes: torch.Tensor, assignments: torch.Tensor, n_labels: int
) -> torch.Tensor:
    # language=rst
    """
    Classify data with the label with highest average spiking activity over all neurons.

    :param spikes: Binary tensor of shape ``(n_samples, time, n_neurons)`` of a layer's
        spiking activity.
    :param assignments: A vector of shape ``(n_neurons,)`` of neuron label assignments.
    :param n_labels: The number of target labels in the data.
    :return: Predictions tensor of shape ``(n_samples,)`` resulting from the "all
        activity" classification scheme. Samples that elicit no activity from any
        assigned neuron are predicted as ``-1`` (abstain) rather than defaulting to
        class ``0``.
    """
    n_samples = spikes.size(0)

    # Sum over time dimension (spike ordering doesn't matter).
    spikes = spikes.sum(1)
    if spikes.is_sparse:
        spikes = spikes.to_dense()

    rates = torch.zeros((n_samples, n_labels), device=spikes.device)
    for i in range(n_labels):
        # Count the number of neurons with this label assignment.
        n_assigns = torch.sum(assignments == i).float()

        if n_assigns > 0:
            # Get indices of samples with this label.
            indices = torch.nonzero(assignments == i).view(-1)

            # Compute layer-wise firing rate for this label.
            rates[:, i] = torch.sum(spikes[:, indices], 1) / n_assigns

    # Predictions are arg-max of layer-wise firing rates.
    predictions = torch.sort(rates, dim=1, descending=True)[1][:, 0]

    # Abstain (-1) on samples with no activity, avoiding a biased vote for class 0.
    predictions[rates.sum(1) == 0] = -1

    return predictions


def proportion_weighting(