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

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 3b49fd9..85301bc 100644
--- a/main.py
+++ b/main.py
@@ -658,7 +658,7 @@ if __name__ == "__main__":
 
         grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=1)
         data = np.array(model.triples['train'])
-        data = np.random.sample(0.2)
+        data = np.random.choice(np.arange(len(data)), size=int(len(data) * 0.2), replace=False)
         dataloader = iter(DataLoader(
                 TrainDataset(data, model.p),
                 batch_size=len(data),