alihan/knowledge-harvest-from-lms
1
0
Fork 0
mirror of https://github.com/tanyuqian/knowledge-harvest-from-lms.git synced 2023-06-02 01:35:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

updated

Browse Source
This commit is contained in:
bwt09
2022-06-18 19:04:10 -07:00
parent f5cac7e9dd
commit 050a6dff23
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
models/knowledge_harvester.py
View File

@@ -127,11 +127,10 @@ class KnowledgeHarvester:
logprobs = self._model.get_mask_filling_logprobs(
prompt=prompt, ent_tuple=ent_tuple)['mask_logprobs']
token_wise_score = sum(logprobs) / len(logprobs)
ent_wise_score = sum(logprobs) / len(ent_tuple)
mean_score = sum(logprobs) / len(logprobs)
min_score = min(logprobs)
return (token_wise_score + ent_wise_score + min_score) / 3.
return (mean_score * 2. + min_score) / 3.
@property
def weighted_ent_tuples(self):