Method loop

methods/pt_map/pt_map_loss.py:104–109

(self, X: torch.Tensor, model: GaussianModel, n_epochs: int = 20)

        model.update_from_estimate(m_estimates, self.alpha)

    def loop(self, X: torch.Tensor, model: GaussianModel, n_epochs: int = 20):
        for epoch in range(1, n_epochs + 1):
            self.perform_epoch(model=model, X=X)
        # get final accuracy and return it
        P = model.get_probas(X=X, labels=self.s_labels)
        return P


class PTMAPLoss(nn.Module):