emanuelaboros commited on
Commit
a5ad8a0
·
1 Parent(s): 7795f59

add comp back

Browse files
Files changed (1) hide show
  1. generic_ner.py +4 -2
generic_ner.py CHANGED
@@ -450,9 +450,11 @@ def postprocess_entities(entities):
450
 
451
  # Step 2: Attach "comp.function" entities to the closest other entities
452
  filtered_entities = attach_comp_to_closest(filtered_entities)
453
- print("After attach_comp_to_closest:", filtered_entities, "\n")
 
454
  filtered_entities = repair_names_in_entities(filtered_entities)
455
- print("After repair_names_in_entities:", filtered_entities, "\n")
 
456
 
457
  # Step 3: Remove entities that are not useful for NEL
458
  # filtered_entities = clean_coarse_entities(filtered_entities)
 
450
 
451
  # Step 2: Attach "comp.function" entities to the closest other entities
452
  filtered_entities = attach_comp_to_closest(filtered_entities)
453
+ if DEBUG:
454
+ print("After attach_comp_to_closest:", filtered_entities, "\n")
455
  filtered_entities = repair_names_in_entities(filtered_entities)
456
+ if DEBUG:
457
+ print("After repair_names_in_entities:", filtered_entities, "\n")
458
 
459
  # Step 3: Remove entities that are not useful for NEL
460
  # filtered_entities = clean_coarse_entities(filtered_entities)