implemented inference

2024-07-04 15:36:01 +02:00
parent 745efbb8f5
commit 5014c41b24
13 changed files with 384 additions and 177 deletions
--- a/Deep-SAD-PyTorch/src/baselines/kde.py
+++ b/Deep-SAD-PyTorch/src/baselines/kde.py
@@ -108,7 +108,9 @@ class KDE(object):
        """Tests the Kernel Density Estimation model on the test data."""
        logger = logging.getLogger()

-        _, test_loader = dataset.loaders(batch_size=128, num_workers=n_jobs_dataloader)
+        _, test_loader, _ = dataset.loaders(
+            batch_size=128, num_workers=n_jobs_dataloader
+        )

        # Get data from loader
        idx_label_score = []