y_true==1 in PadimSVDD

[rishabh-akridata]

Hi @Pangoraw,
What is y_true ==1 in PadimSVDD? Does 1 mean normal image and 0 means anomalous?

pbar = tqdm(range(n_epochs))
        for epoch in pbar:
            loss_epoch = 0
            n_batches = 0
            for imgs, y_true in train_dataloader:
                imgs = imgs.to(self.device)
                y_true = y_true.to(self.device)
                imgs = imgs[y_true == 1]

                optimizer.zero_grad()

Thanks

[Pangoraw]

i am not sure if padimsvdd.pretrain() does something actually. I don't recall exactly what it meant to do

[rishabh-akridata]

Okay.