Source code for avalanche.models.ncm_classifier

import torch
from torch import nn

[docs]class NCMClassifier(nn.Module): """ NCM Classifier. NCMClassifier performs nearest class mean classification measuring the distance between the input tensor and the ones stored in 'self.class_means'. """
[docs] def __init__(self, class_mean=None): """ :param class_mean: tensor of dimension (num_classes x feature_size) used to classify input patterns. """ super().__init__() self.class_means = class_mean
def forward(self, x): pred_inter = (x.T / torch.norm(x.T, dim=0)).T sqd = torch.cdist(self.class_means[:, :].T, pred_inter) return (-sqd).T
__all__ = ["NCMClassifier"]