From bb6d06a903f7e7d731c4903fd5e2215166630970 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Wed, 17 May 2023 13:47:41 +0700
Subject: [PATCH] add grid search

---
 main.py   | 2 +-
 models.py | 6 ------
 2 files changed, 1 insertion(+), 7 deletions(-)

diff --git a/main.py b/main.py
index d1487c5..cfc1667 100644
--- a/main.py
+++ b/main.py
@@ -656,7 +656,7 @@ if __name__ == "__main__":
             'module__embed_dim': [300, 400, 500],
         }
 
-        grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=1)
+        grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=1, scoring=torch.nn.BCELoss)
         data = np.array(model.triples['train'])
         data = data[np.random.choice(np.arange(len(data)), size=int(len(data) * 0.2), replace=False)]
         dataloader = iter(DataLoader(
diff --git a/models.py b/models.py
index a8685c0..90bf38f 100644
--- a/models.py
+++ b/models.py
@@ -537,12 +537,6 @@ class FouriER(torch.nn.Module):
         loss = self.bceloss(pred, true_label)
         return loss
     
-    def score(self, pred, true_label=None, sub_samp=None):
-        label_pos = true_label[0]
-        label_neg = true_label[1:]
-        loss = self.bceloss(pred, true_label)
-        return loss
-    
     def forward_embeddings(self, x):
         x = self.patch_embed(x)
         return x