From 835213ab1b413ce3b6d66eaae81871bb93f90963 Mon Sep 17 00:00:00 2001 From: thanhvc3 Date: Wed, 17 May 2023 13:48:49 +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 cfc1667..5efb746 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, scoring=torch.nn.BCELoss) + grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=3, 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(