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:16:27 +07:00
parent 9aa85307af
commit c8e8d8a660
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -658,7 +658,7 @@ if __name__ == "__main__":
grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=1) grid = GridSearchCV(estimator=estimator, param_grid=paramsGrid, n_jobs=-1, cv=1)
data = np.array(model.triples['train']) 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( dataloader = iter(DataLoader(
TrainDataset(data, model.p), TrainDataset(data, model.p),
batch_size=len(data), batch_size=len(data),