コード例 #1
0
    def extract_same_head_lemma(mention_x: MentionDataLight,
                                mention_y: MentionDataLight) -> RelationType:
        """
        Check if input mentions has same head lemma relation

        Args:
            mention_x: MentionDataLight
            mention_y: MentionDataLight

        Returns:
            RelationType.SAME_HEAD_LEMMA or RelationType.NO_RELATION_FOUND
        """
        if StringUtils.is_preposition(mention_x.mention_head_lemma) or \
                StringUtils.is_preposition(mention_y.mention_head_lemma) or \
                StringUtils.is_determiner(mention_x.mention_head_lemma) or \
                StringUtils.is_determiner(mention_y.mention_head_lemma):
            return RelationType.NO_RELATION_FOUND
        if mention_x.mention_head_lemma == mention_y.mention_head_lemma:
            return RelationType.SAME_HEAD_LEMMA
        return RelationType.NO_RELATION_FOUND
コード例 #2
0
def test_is_determiner():
    assert StringUtils.is_determiner('the')
    assert StringUtils.is_determiner('on') is False
コード例 #3
0
def test_is_determiner():
    assert StringUtils.is_determiner("the")
    assert StringUtils.is_determiner("on") is False