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

---
 main.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/main.py b/main.py
index 24c75e5..2626de1 100644
--- a/main.py
+++ b/main.py
@@ -659,7 +659,6 @@ if __name__ == "__main__":
         grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=1)
         data = np.array(model.triples['train'])
         data = data[np.random.choice(np.arange(len(data)), size=int(len(data) * 0.2), replace=False)]
-        print(data[0])
         dataloader = iter(DataLoader(
                 TrainDataset(data, model.p),
                 batch_size=len(data),
@@ -668,8 +667,6 @@ if __name__ == "__main__":
                 collate_fn=TrainDataset.collate_fn
             ))
         for step, batch in enumerate(dataloader):
-            print(batch[0])
-            print(batch.shape)
             sub, rel, obj, label, neg_ent, sub_samp = model.read_batch(
                 batch, 'train')