<reponame>thunlp/HMEAE

ACE_FILES = './English'
ACE_DUMP = './data/ACE05/'
GloVe_file = './glove/glove.6B.100d.txt'


EVENT_TYPE_TO_ID = {'None': 0, 'Be-Born': 1, 'Die': 2, 'Marry': 3, 'Divorce': 4, 'Injure': 5, 'Transfer-Ownership': 6, 'Transfer-Money': 7, 'Transport': 8, 'Start-Org': 9, 'End-Org': 10, 'Declare-Bankruptcy': 11, 'Merge-Org': 12, 'Attack': 13, 'Demonstrate': 14, 'Meet': 15, 'Phone-Write': 16, 'Start-Position': 17, 'End-Position': 18, 'Nominate': 19, 'Elect': 20, 'Arrest-Jail': 21, 'Release-Parole': 22, 'Charge-Indict': 23, 'Trial-Hearing': 24, 'Sue': 25, 'Convict': 26, 'Sentence': 27, 'Fine': 28, 'Execute': 29, 'Extradite': 30, 'Acquit': 31, 'Pardon': 32, 'Appeal': 33}

ROLE_TO_ID = {'None': 0, 'Person': 1, 'Place': 2, 'Buyer': 3, 'Seller': 4, 'Beneficiary': 5, 'Price': 6, 'Artifact': 7, 'Origin': 8, 'Destination': 9, 'Giver': 10, 'Recipient': 11, 'Money': 12, 'Org': 13, 'Agent': 14, 'Victim': 15, 'Instrument': 16, 'Entity': 17, 'Attacker': 18, 'Target': 19, 'Defendant': 20, 'Adjudicator': 21, 'Prosecutor': 22, 'Plaintiff': 23, 'Crime': 24, 'Position': 25, 'Sentence': 26, 'Vehicle': 27, 'Time-Within': 28, 'Time-Starting': 29, 'Time-Ending': 30, 'Time-Before': 31, 'Time-After': 32, 'Time-Holds': 33, 'Time-At-Beginning': 34, 'Time-At-End': 35}

corenlp_path = './stanford-corenlp-full-2018-10-05'

INF = 1e8

#general hyperparams
embedding_dim = 100
posi_embedding_dim = 5
event_type_embedding_dim = 5

#trigger hyperparameters
t_filters = 200
t_batch_size = 170
t_lr = 0.001
t_epoch = 5
t_keepprob = 0.5

#DMCNN&HMEAE argument hyperparameters
a_filters = 300
a_lr = 0.001
a_keepprob = 0.5
a_epoch = 20
a_batch_size = 20
a_u_c_dim = 900
a_W_a_dim = 900

# Module Design
                                               #Role              Module
module_design     =       [[0,0,0,0,0,0,0,1],  #NA                 NA
                           [1,0,0,0,0,0,0,0],  #Person             Person
                           [0,1,0,0,0,0,0,0],  #Place              Place
                           [1,0,1,0,0,0,0,0],  #Buyer              Person, Org
                           [1,0,1,0,0,0,0,0],  #Seller             Person, Org
                           [1,0,1,0,0,0,0,0],  #Beneficiary        Person, Org
                           [0,0,0,0,1,0,0,0],  #Price              Good
                           [0,0,0,0,0,0,1,0],  #Artifact           Entity
                           [0,1,0,0,0,0,0,0],  #Origin             Place
                           [0,1,0,0,0,0,0,0],  #Destination        Place
                           [1,0,1,0,0,0,0,0],  #Giver              Person, Org
                           [1,0,1,0,0,0,0,0],  #Recipient          Person, Org
                           [0,0,0,0,1,0,0,0],  #Money              Good
                           [0,0,1,0,0,0,0,0],  #Org                Org
                           [1,1,1,0,0,0,0,0],  #Agent              Person, Place, Org
                           [1,0,0,0,0,0,0,0],  #Victim             Person
                           [0,0,0,0,1,0,0,0],  #Instrument         Good
                           [0,0,0,0,0,0,1,0],  #Entity             Entity
                           [1,1,1,0,1,0,0,0],  #Attacker           Person, Org, Place, Good
                           [1,1,1,0,1,0,0,0],  #Target             Person, Org, Place, Good
                           [1,0,1,0,0,0,0,0],  #Defendent          Person, Org
                           [1,0,1,0,0,0,0,0],  #Adjudicator        Person, Org
                           [1,0,1,0,0,0,0,0],  #Prosecutor         Person, Org
                           [1,0,1,0,0,0,0,0],  #Plantiff           Person, Org
                           [0,0,0,0,0,1,0,0],  #Crime              Behavior
                           [0,1,0,0,0,0,0,0],  #Position           Place
                           [0,0,0,0,0,1,0,0],  #Sentence           Behavior
                           [0,0,0,0,1,0,0,0],  #Vehicle            Goods
                           [0,0,0,1,0,0,0,0],  #Time-Within        Time
                           [0,0,0,1,0,0,0,0],  #Time-Starting      Time
                           [0,0,0,1,0,0,0,0],  #Time-Ending        Time
                           [0,0,0,1,0,0,0,0],  #Time-Before        Time
                           [0,0,0,1,0,0,0,0],  #Time-After         Time
                           [0,0,0,1,0,0,0,0],  #Time-Holds         Time
                           [0,0,0,1,0,0,0,0],  #Time-At-Beginning  Time
                           [0,0,0,1,0,0,0,0]]  #Time-At-End        Time

module_num = len(module_design[0])
ACE_FILES = './English'
ACE_DUMP = './data/ACE05/'
GloVe_file = './glove/glove.6B.100d.txt'


EVENT_TYPE_TO_ID = {'None': 0, 'Be-Born': 1, 'Die': 2, 'Marry': 3, 'Divorce': 4, 'Injure': 5, 'Transfer-Ownership': 6, 'Transfer-Money': 7, 'Transport': 8, 'Start-Org': 9, 'End-Org': 10, 'Declare-Bankruptcy': 11, 'Merge-Org': 12, 'Attack': 13, 'Demonstrate': 14, 'Meet': 15, 'Phone-Write': 16, 'Start-Position': 17, 'End-Position': 18, 'Nominate': 19, 'Elect': 20, 'Arrest-Jail': 21, 'Release-Parole': 22, 'Charge-Indict': 23, 'Trial-Hearing': 24, 'Sue': 25, 'Convict': 26, 'Sentence': 27, 'Fine': 28, 'Execute': 29, 'Extradite': 30, 'Acquit': 31, 'Pardon': 32, 'Appeal': 33}

ROLE_TO_ID = {'None': 0, 'Person': 1, 'Place': 2, 'Buyer': 3, 'Seller': 4, 'Beneficiary': 5, 'Price': 6, 'Artifact': 7, 'Origin': 8, 'Destination': 9, 'Giver': 10, 'Recipient': 11, 'Money': 12, 'Org': 13, 'Agent': 14, 'Victim': 15, 'Instrument': 16, 'Entity': 17, 'Attacker': 18, 'Target': 19, 'Defendant': 20, 'Adjudicator': 21, 'Prosecutor': 22, 'Plaintiff': 23, 'Crime': 24, 'Position': 25, 'Sentence': 26, 'Vehicle': 27, 'Time-Within': 28, 'Time-Starting': 29, 'Time-Ending': 30, 'Time-Before': 31, 'Time-After': 32, 'Time-Holds': 33, 'Time-At-Beginning': 34, 'Time-At-End': 35}

corenlp_path = './stanford-corenlp-full-2018-10-05'

INF = 1e8

#general hyperparams
embedding_dim = 100
posi_embedding_dim = 5
event_type_embedding_dim = 5

#trigger hyperparameters
t_filters = 200
t_batch_size = 170
t_lr = 0.001
t_epoch = 5
t_keepprob = 0.5

#DMCNN&HMEAE argument hyperparameters
a_filters = 300
a_lr = 0.001
a_keepprob = 0.5
a_epoch = 20
a_batch_size = 20
a_u_c_dim = 900
a_W_a_dim = 900

# Module Design
                                               #Role              Module
module_design     =       [[0,0,0,0,0,0,0,1],  #NA                 NA
                           [1,0,0,0,0,0,0,0],  #Person             Person
                           [0,1,0,0,0,0,0,0],  #Place              Place
                           [1,0,1,0,0,0,0,0],  #Buyer              Person, Org
                           [1,0,1,0,0,0,0,0],  #Seller             Person, Org
                           [1,0,1,0,0,0,0,0],  #Beneficiary        Person, Org
                           [0,0,0,0,1,0,0,0],  #Price              Good
                           [0,0,0,0,0,0,1,0],  #Artifact           Entity
                           [0,1,0,0,0,0,0,0],  #Origin             Place
                           [0,1,0,0,0,0,0,0],  #Destination        Place
                           [1,0,1,0,0,0,0,0],  #Giver              Person, Org
                           [1,0,1,0,0,0,0,0],  #Recipient          Person, Org
                           [0,0,0,0,1,0,0,0],  #Money              Good
                           [0,0,1,0,0,0,0,0],  #Org                Org
                           [1,1,1,0,0,0,0,0],  #Agent              Person, Place, Org
                           [1,0,0,0,0,0,0,0],  #Victim             Person
                           [0,0,0,0,1,0,0,0],  #Instrument         Good
                           [0,0,0,0,0,0,1,0],  #Entity             Entity
                           [1,1,1,0,1,0,0,0],  #Attacker           Person, Org, Place, Good
                           [1,1,1,0,1,0,0,0],  #Target             Person, Org, Place, Good
                           [1,0,1,0,0,0,0,0],  #Defendent          Person, Org
                           [1,0,1,0,0,0,0,0],  #Adjudicator        Person, Org
                           [1,0,1,0,0,0,0,0],  #Prosecutor         Person, Org
                           [1,0,1,0,0,0,0,0],  #Plantiff           Person, Org
                           [0,0,0,0,0,1,0,0],  #Crime              Behavior
                           [0,1,0,0,0,0,0,0],  #Position           Place
                           [0,0,0,0,0,1,0,0],  #Sentence           Behavior
                           [0,0,0,0,1,0,0,0],  #Vehicle            Goods
                           [0,0,0,1,0,0,0,0],  #Time-Within        Time
                           [0,0,0,1,0,0,0,0],  #Time-Starting      Time
                           [0,0,0,1,0,0,0,0],  #Time-Ending        Time
                           [0,0,0,1,0,0,0,0],  #Time-Before        Time
                           [0,0,0,1,0,0,0,0],  #Time-After         Time
                           [0,0,0,1,0,0,0,0],  #Time-Holds         Time
                           [0,0,0,1,0,0,0,0],  #Time-At-Beginning  Time
                           [0,0,0,1,0,0,0,0]]  #Time-At-End        Time

module_num = len(module_design[0])
#general hyperparams #trigger hyperparameters #DMCNN&HMEAE argument hyperparameters # Module Design #Role              Module #NA                 NA #Person             Person #Place              Place #Buyer              Person, Org #Seller             Person, Org #Beneficiary        Person, Org #Price              Good #Artifact           Entity #Origin             Place #Destination        Place #Giver              Person, Org #Recipient          Person, Org #Money              Good #Org                Org #Agent              Person, Place, Org #Victim             Person #Instrument         Good #Entity             Entity #Attacker           Person, Org, Place, Good #Target             Person, Org, Place, Good #Defendent          Person, Org #Adjudicator        Person, Org #Prosecutor         Person, Org #Plantiff           Person, Org #Crime              Behavior #Position           Place #Sentence           Behavior #Vehicle            Goods #Time-Within        Time #Time-Starting      Time #Time-Ending        Time #Time-Before        Time #Time-After         Time #Time-Holds         Time #Time-At-Beginning  Time #Time-At-End        Time