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:41:13 +07:00
parent 9d182abadb
commit d912e0a225
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
main.py
View File

@ -669,8 +669,13 @@ if __name__ == "__main__":
for step, batch in enumerate(dataloader):
sub, rel, obj, label, neg_ent, sub_samp = model.read_batch(
batch, 'train')
if (neg_ent is None):
neg_ent = np.repeat(None, repeats=len(sub))
else:
neg_ent = neg_ent.cpu()
dataset = np.stack([sub.cpu(), rel.cpu(), neg_ent.cpu(), np.repeat(model.p.train_strategy, repeats=len(sub))], axis = 1)
dataset = np.stack([sub.cpu(), rel.cpu(), neg_ent, np.repeat(model.p.train_strategy, repeats=len(sub))], axis = 1)
search = grid.fit(dataset, label)
print("BEST SCORE: ", search.best_score_)
print("BEST PARAMS: ", search.best_params_)