-
Notifications
You must be signed in to change notification settings - Fork 1
/
entities.py
104 lines (75 loc) · 1.84 KB
/
entities.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
from __future__ import print_function, division
class Entity(object):
def __init__(self, type, id, string, spans, text_id, doctimerel=None, etree=None):
self.type = type
self.id = id
self.string = string
self.spans = spans
self.text_id = text_id
self.doctimerel = doctimerel
self.phi = {}
self.phi_v = None
self.tokens = None
self.paragraph = None
self.xmltree = etree
self.embedding = None
self.next_event = None
self.next_entity = None
self.attributes = {}
def __str__(self):
return str(self.string)
def __hash__(self):
return hash(self.id)
def __eq__(self, other):
return self.id == other.id
def __ne__(self, other):
return not(self == other)
def type(self):
return self.type
def ID(self):
return self.id
def get_tokens(self):
return self.tokens
def text_id(self):
return self.text_id
def get_doctimerel(self):
return self.doctimerel
def get_span(self):
return self.spans[0]
def get_etree(self):
return self.xmltree
def get_doc_id(self):
return self.id.split('@')[2]
class TLink(object):
def __init__(self, e1, e2, tlink=None):
self.e1 = e1
self.e2 = e2
self.tlink = tlink
self.phi = {}
self.phi_v = None
self.tokens_ib = None
self.id = None
def __str__(self):
return str(self.e1) + '-' + str(self.e2)
def ID(self):
if not self.id:
self.id = self.e1.ID() + '-' + self.e2.ID()
return self.id
def __hash__(self):
return hash(self.id())
def __eq__(self, other):
return self.ID() == other.ID()
def __ne__(self, other):
return not (self == other)
def set_tokens_ib(self, tokens):
self.tokens_ib = list(tokens)
def get_tokens_ib(self):
return self.tokens_ib
def type(self):
return self.e1.type + '-' + self.e2.type
def get_tlink(self):
return self.tlink
def get_e1(self):
return self.e1
def get_e2(self):
return self.e2