def extraction_accuracy_calc()

in microservices/extraction_service/src/utils/utils_functions.py [0:0]

• 14 lines of code
• 5 McCabe index (conditional complexity)

def extraction_accuracy_calc(total_entities_list,flag=True):
  """
    This function is to calculate document extraction accuracy
    Parameters
    ----------
    total_entities_list: Total extracted list of dict
    Returns : Extraction score
    -------
  """
  # get fields extraction accuracy
  extraction_status="single entities present"
  if flag is False:
    extraction_accuracy = 0.0
    extraction_status="duplicate entities present"
    return extraction_accuracy,extraction_status
  entity_accuracy_list = [each_entity.get("extraction_confidence") if
                          each_entity.get("extraction_confidence") else 0
                        for each_entity in
                        total_entities_list if not each_entity.
          get("manual_extraction")]

  extraction_accuracy = round(sum(entity_accuracy_list) /
                              len(entity_accuracy_list), 3)

  return extraction_accuracy,extraction_status