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-06 12:15:35 -07:00
parent 595103af34
commit 97f34574bb
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
models/entity_tuple_searcher.py
View File

@@ -160,9 +160,7 @@ class EntityTupleSearcher:
if pred_ent in raw_prompt:
return
ent_logprob = (sum(cur_logprobs) + min(cur_logprobs)) / 2.
heapq.heappush(collected_ent_heap, [ent_logprob, pred_ent])
heapq.heappush(collected_ent_heap, [min(cur_logprobs), pred_ent])
while len(collected_ent_heap) > n:
heapq.heappop(collected_ent_heap)