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 14:04:48 +07:00
parent ddbaa2781f
commit f2052c2839
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
main.py
View File

@ -675,8 +675,14 @@ if __name__ == "__main__":
else: else:
neg_ent = neg_ent.cpu() neg_ent = neg_ent.cpu()
dataset = np.stack([sub.cpu(), rel.cpu(), neg_ent, np.repeat(model.p.train_strategy, repeats=len(sub))], axis = 1) inputs = []
search = grid.fit(dataset, label) for i in len(sub):
input = {}
input['sub'] = sub[i]
input['rel'] = rel[i]
input['neg_ents'] = neg_ent[i]
inputs.append(input)
search = grid.fit(inputs, label)
print("BEST SCORE: ", search.best_score_) print("BEST SCORE: ", search.best_score_)
print("BEST PARAMS: ", search.best_params_) print("BEST PARAMS: ", search.best_params_)
if (args.test_only): if (args.test_only):