cngthnh/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add grid search

Browse Source
This commit is contained in:
thanhvc3
2023-05-17 13:52:52 +07:00
parent 14001cf80d
commit bbbac51e4c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@@ -653,10 +653,10 @@ if __name__ == "__main__":
'optimizer__lr': [0.0001, 0.0003, 0.001],
'optimizer__weight_decay': [1e-4, 1e-5, 1e-6],
'module__hid_drop': [0.2, 0.5, 0.7],
'module__embed_dim': [300, 400, 500],
# 'module__embed_dim': [300, 400, 500],
}
grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=3, scoring=torch.nn.BCELoss)
grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=2, scoring=torch.nn.BCELoss)
data = np.array(model.triples['train'])
data = data[np.random.choice(np.arange(len(data)), size=int(len(data) * 0.15), replace=False)]
dataloader = iter(DataLoader(