BIDI_ML_INFO_EXTRACTION/BiddingKG/deepdive/udf/map_candidate_entity_pair.py

#!/usr/bin/env python
#encoding:utf-8
from deepdive import *
import re

@tsv_extractor
@returns(lambda
        entity1_id       = "text",
        entity1_name     = "text",
        entity2_id           = "text",
        entity2_name  = "text",
    :[])
def extract(
        entity1_id       = "text",
        entity1_name     = "text",
        entity2_id           = "text",
        entity2_name   = "text",
    ):
    """
    Finds phrases that are continuous words tagged with ORG.
    """
    if not(set(entity1_name) <= set(entity2_name) or set(entity2_name) <= set(entity1_name)):
        yield [
            entity1_id,
            entity1_name,
            entity2_id,
            entity2_name,
        ]