ED_LRR/beam_stack_impl.py

7 lines
237 B
Python
Raw Permalink Normal View History

2022-02-23 21:45:59 +00:00
def pruntLayer(l,w):
Keep=sorted(Open[l])[:w] # best `w` node from `Open[l]`
Prune = [n for n in Open[l] if n not in Keep]
beam_stack.top().f_max=min([f(n) for n in Prune])
for n in Prune:
Open[l].remove(n)