├── .gitignore ├── README.md ├── config ├── data_config │ ├── entity │ │ ├── Few-NERD_H1.yaml │ │ ├── Few-NERD_H2.yaml │ │ ├── Few-NERD_H3.yaml │ │ ├── Few-NERD_H4.yaml │ │ ├── Few-NERD_H5.yaml │ │ ├── Few-NERD_H6.yaml │ │ ├── Few-NERD_H7.yaml │ │ ├── Few-NERD_M1.yaml │ │ ├── Few-NERD_M2.yaml │ │ ├── Few-NERD_M3.yaml │ │ ├── Few-NERD_M4.yaml │ │ ├── Few-NERD_M5.yaml │ │ ├── Few-NERD_M6.yaml │ │ ├── Few-NERD_M7.yaml │ │ ├── Few-NERD_R1.yaml │ │ ├── Few-NERD_R2.yaml │ │ ├── Few-NERD_R3.yaml │ │ ├── Few-NERD_R4.yaml │ │ ├── Few-NERD_R5.yaml │ │ ├── Few-NERD_R6.yaml │ │ ├── Few-NERD_R7.yaml │ │ ├── Few-NERD_V1.yaml │ │ ├── Few-NERD_V2.yaml │ │ ├── Few-NERD_V3.yaml │ │ ├── Few-NERD_V4.yaml │ │ ├── Few-NERD_V5.yaml │ │ ├── Few-NERD_V6.yaml │ │ └── Few-NERD_V7.yaml │ ├── event │ │ ├── ace05_event_H1.yaml │ │ ├── ace05_event_H2.yaml │ │ ├── ace05_event_H3.yaml │ │ ├── ace05_event_H4.yaml │ │ ├── ace05_event_H5.yaml │ │ ├── ace05_event_H6.yaml │ │ ├── ace05_event_H7.yaml │ │ ├── ace05_event_M1.yaml │ │ ├── ace05_event_M2.yaml │ │ ├── ace05_event_M3.yaml │ │ ├── ace05_event_M4.yaml │ │ ├── ace05_event_M5.yaml │ │ ├── ace05_event_M6.yaml │ │ ├── ace05_event_M7.yaml │ │ ├── ace05_event_R1.yaml │ │ ├── ace05_event_R2.yaml │ │ ├── ace05_event_R3.yaml │ │ ├── ace05_event_R4.yaml │ │ ├── ace05_event_R5.yaml │ │ ├── ace05_event_R6.yaml │ │ ├── ace05_event_R7.yaml │ │ ├── ace05_event_V1.yaml │ │ ├── ace05_event_V2.yaml │ │ ├── ace05_event_V3.yaml │ │ ├── ace05_event_V4.yaml │ │ ├── ace05_event_V5.yaml │ │ ├── ace05_event_V6.yaml │ │ └── ace05_event_V7.yaml │ └── relation │ │ ├── NYT_H1.yaml │ │ ├── NYT_H2.yaml │ │ ├── NYT_H3.yaml │ │ ├── NYT_H4.yaml │ │ ├── NYT_H5.yaml │ │ ├── NYT_H6.yaml │ │ ├── NYT_H7.yaml │ │ ├── NYT_M1.yaml │ │ ├── NYT_M2.yaml │ │ ├── NYT_M3.yaml │ │ ├── NYT_M4.yaml │ │ ├── NYT_M5.yaml │ │ ├── NYT_M6.yaml │ │ ├── NYT_M7.yaml │ │ ├── NYT_R1.yaml │ │ ├── NYT_R2.yaml │ │ ├── NYT_R3.yaml │ │ ├── NYT_R4.yaml │ │ ├── NYT_R5.yaml │ │ ├── NYT_R6.yaml │ │ ├── NYT_R7.yaml │ │ ├── NYT_V1.yaml │ │ ├── NYT_V2.yaml │ │ ├── NYT_V3.yaml │ │ ├── NYT_V4.yaml │ │ ├── NYT_V5.yaml │ │ ├── NYT_V6.yaml │ │ └── NYT_V7.yaml ├── offset_map │ ├── closest_offset_en.yaml │ ├── first_offset_en.yaml │ └── longer_first_offset_en.yaml └── prompt_conf │ ├── Few-NERD.ini │ ├── Few-NERD_H.ini │ ├── Few-NERD_M.ini │ ├── Few-NERD_V.ini │ ├── NYT.ini │ ├── NYT_H.ini │ ├── NYT_M.ini │ ├── NYT_V.ini │ ├── ace05_event.ini │ ├── ace05_event_H.ini │ ├── ace05_event_M.ini │ ├── ace05_event_R.ini │ └── ace05_event_V.ini ├── dataset_construct ├── .gitignore ├── README.md ├── data_convert.py ├── scripts │ ├── process_nerd.py │ └── run_data_convert.bash └── universal_ie │ ├── __init__.py │ ├── dataset.py │ ├── ie_format.py │ ├── logger.py │ ├── record_schema.py │ ├── task_format │ ├── __init__.py │ ├── jointer.py │ ├── nerd.py │ ├── oneie.py │ └── task_format.py │ └── utils.py ├── etc └── record.dataload.schema ├── inference.py ├── inference_mul.py ├── requirements.txt ├── run_prompt.py ├── scripts ├── fine_prompt.bash ├── run_noprompt.bash └── run_prompt.bash ├── uie ├── __init__.py ├── extraction │ ├── __init__.py │ ├── constants.py │ ├── dataset_processer.py │ ├── extraction_metrics.py │ ├── label_tree.py │ ├── noiser │ │ ├── __init__.py │ │ └── spot_asoc_noiser.py │ ├── predict_parser │ │ ├── __init__.py │ │ ├── predict_parser.py │ │ ├── spotasoc_predict_parser.py │ │ └── utils.py │ ├── record_schema.py │ ├── scorer.py │ └── utils.py ├── sel2record │ ├── __init__.py │ ├── record.py │ ├── sel2record.py │ └── text2spotasoc.py └── seq2seq │ ├── __init__.py │ ├── constrained_seq2seq.py │ ├── constrained_seq2seq_prompt.py │ ├── constraint_decoder │ ├── __init__.py │ ├── constraint_decoder.py │ └── spotasoc_constraint_decoder.py │ ├── data_collator │ ├── __init__.py │ └── meta_data_collator.py │ ├── features.py │ ├── modeling_t5.py │ ├── models.py │ ├── t5_bert_tokenizer.py │ └── trainer_arguments.py └── uie_json.py /.gitignore: -------------------------------------------------------------------------------- 1 | # UIE 2 | /data 3 | /dataset_processing/data 4 | /dataset_processing/log 5 | /wandb 6 | /pretrain_data 7 | /output 8 | /output_eval 9 | /output_infer 10 | /hf_models 11 | /pd_models 12 | /runs 13 | /models 14 | *.lock 15 | 16 | # mac 17 | .DS_Store 18 | 19 | # env 20 | /.vscode 21 | /.idea 22 | **/__pycache__ 23 | *.pyc 24 | .pytest_cache 25 | -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M1.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | delete_list: 11 | - art-broadcastprogram 12 | - art-film 13 | - art-music 14 | - art-other 15 | - art-painting 16 | - art-writtenart 17 | - person-actor 18 | - person-artist/author 19 | - person-athlete 20 | - person-director 21 | - person-other 22 | - person-politician 23 | - person-scholar 24 | - person-soldier 25 | mapper: 26 | O: O 27 | art-broadcastprogram: broadcastprogram 28 | art-film: film 29 | art-music: music 30 | art-other: art other 31 | art-painting: painting 32 | art-writtenart: writtenart 33 | building-airport: building 34 | building-hospital: building 35 | building-hotel: building 36 | building-library: building 37 | building-other: building 38 | building-restaurant: building 39 | building-sportsfacility: building 40 | building-theater: theater 41 | event-attack/battle/war/militaryconflict: attack 42 | event-disaster: event 43 | event-election: event 44 | event-other: event other 45 | event-protest: event 46 | event-sportsevent: sportsevent 47 | location-GPE: geographical social political 48 | location-bodiesofwater: bodiesofwater 49 | location-island: location 50 | location-mountain: location 51 | location-other: location other 52 | location-park: park 53 | location-road/railway/highway/transit: location 54 | organization-company: organization 55 | organization-education: education 56 | organization-government/governmentagency: organization 57 | organization-media/newspaper: media 58 | organization-other: organization other 59 | organization-politicalparty: organization 60 | organization-religion: organization 61 | organization-showorganization: organization 62 | organization-sportsleague: organization 63 | organization-sportsteam: organization 64 | other-astronomything: astronomything 65 | other-award: award 66 | other-biologything: biologything 67 | other-chemicalthing: chemicalthing 68 | other-currency: currency 69 | other-disease: disease 70 | other-educationaldegree: educationaldegree 71 | other-god: god 72 | other-language: language 73 | other-law: law 74 | other-livingthing: livingthing 75 | other-medical: medical 76 | person-actor: actor 77 | person-artist/author: artist 78 | person-athlete: athlete 79 | person-director: director 80 | person-other: person other 81 | person-politician: politician 82 | person-scholar: scholar 83 | person-soldier: soldier 84 | product-airplane: airplane 85 | product-car: car 86 | product-food: food 87 | product-game: game 88 | product-other: product other 89 | product-ship: ship 90 | product-software: product 91 | product-train: product 92 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M2.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - disaster 11 | - election 12 | - library 13 | - person 14 | - person other 15 | - protest 16 | - train 17 | delete_list: 18 | - art-broadcastprogram 19 | - art-film 20 | - art-music 21 | - art-other 22 | - art-painting 23 | - art-writtenart 24 | mapper: 25 | O: O 26 | art-broadcastprogram: broadcastprogram 27 | art-film: film 28 | art-music: music 29 | art-other: art other 30 | art-painting: painting 31 | art-writtenart: writtenart 32 | building-airport: building 33 | building-hospital: building 34 | building-hotel: building 35 | building-library: library 36 | building-other: building 37 | building-restaurant: building 38 | building-sportsfacility: building 39 | building-theater: theater 40 | event-attack/battle/war/militaryconflict: attack 41 | event-disaster: disaster 42 | event-election: election 43 | event-other: event other 44 | event-protest: protest 45 | event-sportsevent: sportsevent 46 | location-GPE: geographical social political 47 | location-bodiesofwater: bodiesofwater 48 | location-island: location 49 | location-mountain: location 50 | location-other: location other 51 | location-park: park 52 | location-road/railway/highway/transit: location 53 | organization-company: organization 54 | organization-education: education 55 | organization-government/governmentagency: organization 56 | organization-media/newspaper: media 57 | organization-other: organization other 58 | organization-politicalparty: organization 59 | organization-religion: organization 60 | organization-showorganization: organization 61 | organization-sportsleague: organization 62 | organization-sportsteam: organization 63 | other-astronomything: astronomything 64 | other-award: award 65 | other-biologything: biologything 66 | other-chemicalthing: chemicalthing 67 | other-currency: currency 68 | other-disease: disease 69 | other-educationaldegree: educationaldegree 70 | other-god: god 71 | other-language: language 72 | other-law: law 73 | other-livingthing: livingthing 74 | other-medical: medical 75 | person-actor: person 76 | person-artist/author: person 77 | person-athlete: person 78 | person-director: person 79 | person-other: person other 80 | person-politician: person 81 | person-scholar: person 82 | person-soldier: person 83 | product-airplane: airplane 84 | product-car: car 85 | product-food: food 86 | product-game: game 87 | product-other: product other 88 | product-ship: ship 89 | product-software: product 90 | product-train: train 91 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M3.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - athlete 12 | - building other 13 | - disaster 14 | - election 15 | - library 16 | - person 17 | - person other 18 | - protest 19 | - restaurant 20 | - road 21 | - sportsfacility 22 | - train 23 | delete_list: 24 | - art-broadcastprogram 25 | - art-film 26 | - art-music 27 | - art-other 28 | - art-painting 29 | - art-writtenart 30 | mapper: 31 | O: O 32 | art-broadcastprogram: broadcastprogram 33 | art-film: film 34 | art-music: music 35 | art-other: art other 36 | art-painting: painting 37 | art-writtenart: writtenart 38 | building-airport: building 39 | building-hospital: building 40 | building-hotel: building 41 | building-library: library 42 | building-other: building other 43 | building-restaurant: restaurant 44 | building-sportsfacility: sportsfacility 45 | building-theater: theater 46 | event-attack/battle/war/militaryconflict: attack 47 | event-disaster: disaster 48 | event-election: election 49 | event-other: event other 50 | event-protest: protest 51 | event-sportsevent: sportsevent 52 | location-GPE: geographical social political 53 | location-bodiesofwater: bodiesofwater 54 | location-island: location 55 | location-mountain: location 56 | location-other: location other 57 | location-park: park 58 | location-road/railway/highway/transit: road 59 | organization-company: organization 60 | organization-education: education 61 | organization-government/governmentagency: organization 62 | organization-media/newspaper: media 63 | organization-other: organization other 64 | organization-politicalparty: organization 65 | organization-religion: organization 66 | organization-showorganization: organization 67 | organization-sportsleague: organization 68 | organization-sportsteam: organization 69 | other-astronomything: astronomything 70 | other-award: award 71 | other-biologything: biologything 72 | other-chemicalthing: chemicalthing 73 | other-currency: currency 74 | other-disease: disease 75 | other-educationaldegree: educationaldegree 76 | other-god: god 77 | other-language: language 78 | other-law: law 79 | other-livingthing: livingthing 80 | other-medical: medical 81 | person-actor: actor 82 | person-artist/author: person 83 | person-athlete: athlete 84 | person-director: person 85 | person-other: person other 86 | person-politician: person 87 | person-scholar: person 88 | person-soldier: person 89 | product-airplane: airplane 90 | product-car: car 91 | product-food: food 92 | product-game: game 93 | product-other: product other 94 | product-ship: ship 95 | product-software: product 96 | product-train: train 97 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M4.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art 13 | - athlete 14 | - building other 15 | - disaster 16 | - election 17 | - hospital 18 | - library 19 | - person 20 | - person other 21 | - politician 22 | - protest 23 | - restaurant 24 | - road 25 | - scholar 26 | - soldier 27 | - sportsfacility 28 | - train 29 | - writtenart 30 | delete_list: 31 | mapper: 32 | O: O 33 | art-broadcastprogram: art 34 | art-film: art 35 | art-music: art 36 | art-other: art 37 | art-painting: art 38 | art-writtenart: writtenart 39 | building-airport: airport 40 | building-hospital: hospital 41 | building-hotel: building 42 | building-library: library 43 | building-other: building other 44 | building-restaurant: restaurant 45 | building-sportsfacility: sportsfacility 46 | building-theater: theater 47 | event-attack/battle/war/militaryconflict: attack 48 | event-disaster: disaster 49 | event-election: election 50 | event-other: event other 51 | event-protest: protest 52 | event-sportsevent: sportsevent 53 | location-GPE: geographical social political 54 | location-bodiesofwater: bodiesofwater 55 | location-island: location 56 | location-mountain: location 57 | location-other: location other 58 | location-park: park 59 | location-road/railway/highway/transit: road 60 | organization-company: organization 61 | organization-education: education 62 | organization-government/governmentagency: organization 63 | organization-media/newspaper: media 64 | organization-other: organization other 65 | organization-politicalparty: organization 66 | organization-religion: organization 67 | organization-showorganization: organization 68 | organization-sportsleague: organization 69 | organization-sportsteam: organization 70 | other-astronomything: astronomything 71 | other-award: award 72 | other-biologything: biologything 73 | other-chemicalthing: chemicalthing 74 | other-currency: currency 75 | other-disease: disease 76 | other-educationaldegree: educationaldegree 77 | other-god: god 78 | other-language: language 79 | other-law: law 80 | other-livingthing: livingthing 81 | other-medical: medical 82 | person-actor: actor 83 | person-artist/author: person 84 | person-athlete: athlete 85 | person-director: person 86 | person-other: person other 87 | person-politician: politician 88 | person-scholar: scholar 89 | person-soldier: soldier 90 | product-airplane: airplane 91 | product-car: car 92 | product-food: food 93 | product-game: game 94 | product-other: product other 95 | product-ship: ship 96 | product-software: product 97 | product-train: train 98 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M5.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art 13 | - art other 14 | - athlete 15 | - building other 16 | - disaster 17 | - election 18 | - government 19 | - hospital 20 | - library 21 | - music 22 | - painting 23 | - person 24 | - person other 25 | - politician 26 | - protest 27 | - restaurant 28 | - road 29 | - scholar 30 | - soldier 31 | - sportsfacility 32 | - sportsleague 33 | - sportsteam 34 | - train 35 | - writtenart 36 | delete_list: 37 | mapper: 38 | O: O 39 | art-broadcastprogram: art 40 | art-film: art 41 | art-music: music 42 | art-other: art other 43 | art-painting: painting 44 | art-writtenart: writtenart 45 | building-airport: airport 46 | building-hospital: hospital 47 | building-hotel: building 48 | building-library: library 49 | building-other: building other 50 | building-restaurant: restaurant 51 | building-sportsfacility: sportsfacility 52 | building-theater: theater 53 | event-attack/battle/war/militaryconflict: attack 54 | event-disaster: disaster 55 | event-election: election 56 | event-other: event other 57 | event-protest: protest 58 | event-sportsevent: sportsevent 59 | location-GPE: geographical social political 60 | location-bodiesofwater: bodiesofwater 61 | location-island: location 62 | location-mountain: location 63 | location-other: location other 64 | location-park: park 65 | location-road/railway/highway/transit: road 66 | organization-company: organization 67 | organization-education: education 68 | organization-government/governmentagency: government 69 | organization-media/newspaper: media 70 | organization-other: organization other 71 | organization-politicalparty: organization 72 | organization-religion: organization 73 | organization-showorganization: organization 74 | organization-sportsleague: sportsleague 75 | organization-sportsteam: sportsteam 76 | other-astronomything: astronomything 77 | other-award: award 78 | other-biologything: biologything 79 | other-chemicalthing: chemicalthing 80 | other-currency: currency 81 | other-disease: disease 82 | other-educationaldegree: educationaldegree 83 | other-god: god 84 | other-language: language 85 | other-law: law 86 | other-livingthing: livingthing 87 | other-medical: medical 88 | person-actor: actor 89 | person-artist/author: person 90 | person-athlete: athlete 91 | person-director: person 92 | person-other: person other 93 | person-politician: politician 94 | person-scholar: scholar 95 | person-soldier: soldier 96 | product-airplane: airplane 97 | product-car: car 98 | product-food: food 99 | product-game: game 100 | product-other: product other 101 | product-ship: ship 102 | product-software: product 103 | product-train: train 104 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M6.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art other 13 | - athlete 14 | - broadcastprogram 15 | - building other 16 | - company 17 | - disaster 18 | - election 19 | - film 20 | - government 21 | - hospital 22 | - hotel 23 | - island 24 | - library 25 | - mountain 26 | - music 27 | - painting 28 | - person 29 | - person other 30 | - politician 31 | - protest 32 | - restaurant 33 | - road 34 | - scholar 35 | - soldier 36 | - sportsfacility 37 | - sportsleague 38 | - sportsteam 39 | - train 40 | - writtenart 41 | delete_list: 42 | mapper: 43 | O: O 44 | art-broadcastprogram: broadcastprogram 45 | art-film: film 46 | art-music: music 47 | art-other: art other 48 | art-painting: painting 49 | art-writtenart: writtenart 50 | building-airport: airport 51 | building-hospital: hospital 52 | building-hotel: hotel 53 | building-library: library 54 | building-other: building other 55 | building-restaurant: restaurant 56 | building-sportsfacility: sportsfacility 57 | building-theater: theater 58 | event-attack/battle/war/militaryconflict: attack 59 | event-disaster: disaster 60 | event-election: election 61 | event-other: event other 62 | event-protest: protest 63 | event-sportsevent: sportsevent 64 | location-GPE: geographical social political 65 | location-bodiesofwater: bodiesofwater 66 | location-island: island 67 | location-mountain: mountain 68 | location-other: location other 69 | location-park: park 70 | location-road/railway/highway/transit: road 71 | organization-company: company 72 | organization-education: education 73 | organization-government/governmentagency: government 74 | organization-media/newspaper: media 75 | organization-other: organization other 76 | organization-politicalparty: organization 77 | organization-religion: organization 78 | organization-showorganization: organization 79 | organization-sportsleague: sportsleague 80 | organization-sportsteam: sportsteam 81 | other-astronomything: astronomything 82 | other-award: award 83 | other-biologything: biologything 84 | other-chemicalthing: chemicalthing 85 | other-currency: currency 86 | other-disease: disease 87 | other-educationaldegree: educationaldegree 88 | other-god: god 89 | other-language: language 90 | other-law: law 91 | other-livingthing: livingthing 92 | other-medical: medical 93 | person-actor: actor 94 | person-artist/author: person 95 | person-athlete: athlete 96 | person-director: person 97 | person-other: person other 98 | person-politician: politician 99 | person-scholar: scholar 100 | person-soldier: soldier 101 | product-airplane: airplane 102 | product-car: car 103 | product-food: food 104 | product-game: game 105 | product-other: product other 106 | product-ship: ship 107 | product-software: product 108 | product-train: train 109 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_M7.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art other 13 | - artist 14 | - athlete 15 | - broadcastprogram 16 | - building other 17 | - company 18 | - director 19 | - disaster 20 | - election 21 | - film 22 | - government 23 | - hospital 24 | - hotel 25 | - island 26 | - library 27 | - mountain 28 | - music 29 | - painting 30 | - person other 31 | - politicalparty 32 | - politician 33 | - protest 34 | - religion 35 | - restaurant 36 | - road 37 | - scholar 38 | - showorganization 39 | - software 40 | - soldier 41 | - sportsfacility 42 | - sportsleague 43 | - sportsteam 44 | - train 45 | - writtenart 46 | delete_list: 47 | mapper: 48 | O: O 49 | art-broadcastprogram: broadcastprogram 50 | art-film: film 51 | art-music: music 52 | art-other: art other 53 | art-painting: painting 54 | art-writtenart: writtenart 55 | building-airport: airport 56 | building-hospital: hospital 57 | building-hotel: hotel 58 | building-library: library 59 | building-other: building other 60 | building-restaurant: restaurant 61 | building-sportsfacility: sportsfacility 62 | building-theater: theater 63 | event-attack/battle/war/militaryconflict: attack 64 | event-disaster: disaster 65 | event-election: election 66 | event-other: event other 67 | event-protest: protest 68 | event-sportsevent: sportsevent 69 | location-GPE: geographical social political 70 | location-bodiesofwater: bodiesofwater 71 | location-island: island 72 | location-mountain: mountain 73 | location-other: location other 74 | location-park: park 75 | location-road/railway/highway/transit: road 76 | organization-company: company 77 | organization-education: education 78 | organization-government/governmentagency: government 79 | organization-media/newspaper: media 80 | organization-other: organization other 81 | organization-politicalparty: politicalparty 82 | organization-religion: religion 83 | organization-showorganization: showorganization 84 | organization-sportsleague: sportsleague 85 | organization-sportsteam: sportsteam 86 | other-astronomything: astronomything 87 | other-award: award 88 | other-biologything: biologything 89 | other-chemicalthing: chemicalthing 90 | other-currency: currency 91 | other-disease: disease 92 | other-educationaldegree: educationaldegree 93 | other-god: god 94 | other-language: language 95 | other-law: law 96 | other-livingthing: livingthing 97 | other-medical: medical 98 | person-actor: actor 99 | person-artist/author: artist 100 | person-athlete: athlete 101 | person-director: director 102 | person-other: person other 103 | person-politician: politician 104 | person-scholar: scholar 105 | person-soldier: soldier 106 | product-airplane: airplane 107 | product-car: car 108 | product-food: food 109 | product-game: game 110 | product-other: product other 111 | product-ship: ship 112 | product-software: software 113 | product-train: train 114 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_R1.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | delete_list: 11 | mapper: 12 | O: O 13 | art-broadcastprogram: broadcastprogram 14 | art-film: film 15 | art-music: music 16 | art-other: art other 17 | art-painting: painting 18 | art-writtenart: writtenart 19 | building-airport: airport 20 | building-hospital: hospital 21 | building-hotel: hotel 22 | building-library: library 23 | building-other: building other 24 | building-restaurant: restaurant 25 | building-sportsfacility: sportsfacility 26 | building-theater: theater 27 | event-attack/battle/war/militaryconflict: attack 28 | event-disaster: disaster 29 | event-election: election 30 | event-other: event other 31 | event-protest: protest 32 | event-sportsevent: sportsevent 33 | location-GPE: geographical social political 34 | location-bodiesofwater: bodiesofwater 35 | location-island: island 36 | location-mountain: mountain 37 | location-other: location other 38 | location-park: park 39 | location-road/railway/highway/transit: road 40 | organization-company: company 41 | organization-education: education 42 | organization-government/governmentagency: government 43 | organization-media/newspaper: media 44 | organization-other: organization other 45 | organization-politicalparty: politicalparty 46 | organization-religion: religion 47 | organization-showorganization: showorganization 48 | organization-sportsleague: sportsleague 49 | organization-sportsteam: sportsteam 50 | other-astronomything: astronomything 51 | other-award: award 52 | other-biologything: biologything 53 | other-chemicalthing: chemicalthing 54 | other-currency: currency 55 | other-disease: disease 56 | other-educationaldegree: educationaldegree 57 | other-god: god 58 | other-language: language 59 | other-law: law 60 | other-livingthing: livingthing 61 | other-medical: medical 62 | person-actor: actor 63 | person-artist/author: artist 64 | person-athlete: athlete 65 | person-director: director 66 | person-other: person other 67 | person-politician: politician 68 | person-scholar: scholar 69 | person-soldier: soldier 70 | product-airplane: airplane 71 | product-car: car 72 | product-food: food 73 | product-game: game 74 | product-other: product other 75 | product-ship: ship 76 | product-software: software 77 | product-train: train 78 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_R2.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - isle 11 | - newspaper 12 | - biologic 13 | - hill 14 | - statesman 15 | - governmentagency 16 | delete_list: 17 | mapper: 18 | O: O 19 | art-broadcastprogram: broadcastprogram 20 | art-film: film 21 | art-music: music 22 | art-other: art other 23 | art-painting: painting 24 | art-writtenart: writtenart 25 | building-airport: airport 26 | building-hospital: hospital 27 | building-hotel: hotel 28 | building-library: library 29 | building-other: building other 30 | building-restaurant: restaurant 31 | building-sportsfacility: sportsfacility 32 | building-theater: theater 33 | event-attack/battle/war/militaryconflict: attack 34 | event-disaster: disaster 35 | event-election: election 36 | event-other: event other 37 | event-protest: protest 38 | event-sportsevent: sportsevent 39 | location-GPE: geographical social political 40 | location-bodiesofwater: bodiesofwater 41 | location-island: isle 42 | location-mountain: hill 43 | location-other: location other 44 | location-park: park 45 | location-road/railway/highway/transit: road 46 | organization-company: company 47 | organization-education: education 48 | organization-government/governmentagency: governmentagency 49 | organization-media/newspaper: newspaper 50 | organization-other: organization other 51 | organization-politicalparty: politicalparty 52 | organization-religion: religion 53 | organization-showorganization: showorganization 54 | organization-sportsleague: sportsleague 55 | organization-sportsteam: sportsteam 56 | other-astronomything: astronomything 57 | other-award: award 58 | other-biologything: biologic 59 | other-chemicalthing: chemicalthing 60 | other-currency: currency 61 | other-disease: disease 62 | other-educationaldegree: educationaldegree 63 | other-god: god 64 | other-language: language 65 | other-law: law 66 | other-livingthing: livingthing 67 | other-medical: medical 68 | person-actor: actor 69 | person-artist/author: artist 70 | person-athlete: athlete 71 | person-director: director 72 | person-other: person other 73 | person-politician: statesman 74 | person-scholar: scholar 75 | person-soldier: soldier 76 | product-airplane: airplane 77 | product-car: car 78 | product-food: food 79 | product-game: game 80 | product-other: product other 81 | product-ship: ship 82 | product-software: software 83 | product-train: train 84 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_R3.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - performer 11 | - isle 12 | - newspaper 13 | - program 14 | - biologic 15 | - academic qualification 16 | - hill 17 | - organization else 18 | - aircraft 19 | - belief 20 | - statesman 21 | - governmentagency 22 | delete_list: 23 | mapper: 24 | O: O 25 | art-broadcastprogram: broadcastprogram 26 | art-film: film 27 | art-music: music 28 | art-other: art other 29 | art-painting: painting 30 | art-writtenart: writtenart 31 | building-airport: airport 32 | building-hospital: hospital 33 | building-hotel: hotel 34 | building-library: library 35 | building-other: building other 36 | building-restaurant: restaurant 37 | building-sportsfacility: sportsfacility 38 | building-theater: theater 39 | event-attack/battle/war/militaryconflict: attack 40 | event-disaster: disaster 41 | event-election: election 42 | event-other: event other 43 | event-protest: protest 44 | event-sportsevent: sportsevent 45 | location-GPE: geographical social political 46 | location-bodiesofwater: bodiesofwater 47 | location-island: isle 48 | location-mountain: hill 49 | location-other: location other 50 | location-park: park 51 | location-road/railway/highway/transit: road 52 | organization-company: company 53 | organization-education: education 54 | organization-government/governmentagency: governmentagency 55 | organization-media/newspaper: newspaper 56 | organization-other: organization else 57 | organization-politicalparty: politicalparty 58 | organization-religion: religion 59 | organization-showorganization: showorganization 60 | organization-sportsleague: sportsleague 61 | organization-sportsteam: sportsteam 62 | other-astronomything: astronomything 63 | other-award: award 64 | other-biologything: biologic 65 | other-chemicalthing: chemicalthing 66 | other-currency: currency 67 | other-disease: disease 68 | other-educationaldegree: academic qualification 69 | other-god: belief 70 | other-language: language 71 | other-law: law 72 | other-livingthing: livingthing 73 | other-medical: medical 74 | person-actor: performer 75 | person-artist/author: artist 76 | person-athlete: athlete 77 | person-director: director 78 | person-other: person other 79 | person-politician: statesman 80 | person-scholar: scholar 81 | person-soldier: soldier 82 | product-airplane: aircraft 83 | product-car: car 84 | product-food: food 85 | product-game: game 86 | product-other: product other 87 | product-ship: ship 88 | product-software: program 89 | product-train: train 90 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_R4.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - performer 11 | - parliamentary 12 | - faith 13 | - isle 14 | - sports program 15 | - newspaper 16 | - chemistry 17 | - program 18 | - biologic 19 | - academic qualification 20 | - hill 21 | - drawing 22 | - building othergates 23 | - organization else 24 | - aircraft 25 | - belief 26 | - statesman 27 | - governmentagency 28 | delete_list: 29 | mapper: 30 | O: O 31 | art-broadcastprogram: broadcastprogram 32 | art-film: film 33 | art-music: music 34 | art-other: art other 35 | art-painting: drawing 36 | art-writtenart: writtenart 37 | building-airport: airport 38 | building-hospital: hospital 39 | building-hotel: hotel 40 | building-library: library 41 | building-other: building othergates 42 | building-restaurant: restaurant 43 | building-sportsfacility: sportsfacility 44 | building-theater: theater 45 | event-attack/battle/war/militaryconflict: attack 46 | event-disaster: disaster 47 | event-election: election 48 | event-other: event other 49 | event-protest: protest 50 | event-sportsevent: sports program 51 | location-GPE: geographical social political 52 | location-bodiesofwater: bodiesofwater 53 | location-island: isle 54 | location-mountain: hill 55 | location-other: location other 56 | location-park: park 57 | location-road/railway/highway/transit: road 58 | organization-company: company 59 | organization-education: education 60 | organization-government/governmentagency: governmentagency 61 | organization-media/newspaper: newspaper 62 | organization-other: organization else 63 | organization-politicalparty: parliamentary 64 | organization-religion: faith 65 | organization-showorganization: showorganization 66 | organization-sportsleague: sportsleague 67 | organization-sportsteam: sportsteam 68 | other-astronomything: astronomything 69 | other-award: award 70 | other-biologything: biologic 71 | other-chemicalthing: chemistry 72 | other-currency: currency 73 | other-disease: disease 74 | other-educationaldegree: academic qualification 75 | other-god: belief 76 | other-language: language 77 | other-law: law 78 | other-livingthing: livingthing 79 | other-medical: medical 80 | person-actor: performer 81 | person-artist/author: artist 82 | person-athlete: athlete 83 | person-director: director 84 | person-other: person other 85 | person-politician: statesman 86 | person-scholar: scholar 87 | person-soldier: soldier 88 | product-airplane: aircraft 89 | product-car: car 90 | product-food: food 91 | product-game: game 92 | product-other: product other 93 | product-ship: ship 94 | product-software: program 95 | product-train: train 96 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_R5.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - faith 11 | - isle 12 | - autocar 13 | - chemistry 14 | - battle 15 | - belief 16 | - governmentagency 17 | - parliamentary 18 | - gymnasium 19 | - program 20 | - foodstuff 21 | - author 22 | - drawing 23 | - building othergates 24 | - organization else 25 | - performer 26 | - aircraft 27 | - sports program 28 | - newspaper 29 | - biologic 30 | - academic qualification 31 | - hill 32 | - product otherness 33 | - statesman 34 | delete_list: 35 | mapper: 36 | O: O 37 | art-broadcastprogram: broadcastprogram 38 | art-film: film 39 | art-music: music 40 | art-other: art other 41 | art-painting: drawing 42 | art-writtenart: writtenart 43 | building-airport: airport 44 | building-hospital: hospital 45 | building-hotel: hotel 46 | building-library: library 47 | building-other: building othergates 48 | building-restaurant: restaurant 49 | building-sportsfacility: gymnasium 50 | building-theater: theater 51 | event-attack/battle/war/militaryconflict: battle 52 | event-disaster: disaster 53 | event-election: election 54 | event-other: event other 55 | event-protest: protest 56 | event-sportsevent: sports program 57 | location-GPE: geographical social political 58 | location-bodiesofwater: bodiesofwater 59 | location-island: isle 60 | location-mountain: hill 61 | location-other: location other 62 | location-park: park 63 | location-road/railway/highway/transit: road 64 | organization-company: company 65 | organization-education: education 66 | organization-government/governmentagency: governmentagency 67 | organization-media/newspaper: newspaper 68 | organization-other: organization else 69 | organization-politicalparty: parliamentary 70 | organization-religion: faith 71 | organization-showorganization: showorganization 72 | organization-sportsleague: sportsleague 73 | organization-sportsteam: sportsteam 74 | other-astronomything: astronomything 75 | other-award: award 76 | other-biologything: biologic 77 | other-chemicalthing: chemistry 78 | other-currency: currency 79 | other-disease: disease 80 | other-educationaldegree: academic qualification 81 | other-god: belief 82 | other-language: language 83 | other-law: law 84 | other-livingthing: livingthing 85 | other-medical: medical 86 | person-actor: performer 87 | person-artist/author: author 88 | person-athlete: athlete 89 | person-director: director 90 | person-other: person other 91 | person-politician: statesman 92 | person-scholar: scholar 93 | person-soldier: soldier 94 | product-airplane: aircraft 95 | product-car: autocar 96 | product-food: foodstuff 97 | product-game: game 98 | product-other: product otherness 99 | product-ship: ship 100 | product-software: program 101 | product-train: train 102 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_R6.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - faith 11 | - isle 12 | - autocar 13 | - chemistry 14 | - other part 15 | - battle 16 | - belief 17 | - governmentagency 18 | - parliamentary 19 | - clinic 20 | - gymnasium 21 | - aerodrome 22 | - program 23 | - foodstuff 24 | - author 25 | - drawing 26 | - building othergates 27 | - organization else 28 | - performer 29 | - aircraft 30 | - sportsman 31 | - song 32 | - sports program 33 | - newspaper 34 | - biologic 35 | - academic qualification 36 | - hill 37 | - lake 38 | - product otherness 39 | - statesman 40 | delete_list: 41 | mapper: 42 | O: O 43 | art-broadcastprogram: broadcastprogram 44 | art-film: film 45 | art-music: song 46 | art-other: art other 47 | art-painting: drawing 48 | art-writtenart: writtenart 49 | building-airport: aerodrome 50 | building-hospital: clinic 51 | building-hotel: hotel 52 | building-library: library 53 | building-other: building othergates 54 | building-restaurant: restaurant 55 | building-sportsfacility: gymnasium 56 | building-theater: theater 57 | event-attack/battle/war/militaryconflict: battle 58 | event-disaster: disaster 59 | event-election: election 60 | event-other: event other 61 | event-protest: protest 62 | event-sportsevent: sports program 63 | location-GPE: geographical social political 64 | location-bodiesofwater: lake 65 | location-island: isle 66 | location-mountain: hill 67 | location-other: other part 68 | location-park: park 69 | location-road/railway/highway/transit: road 70 | organization-company: company 71 | organization-education: education 72 | organization-government/governmentagency: governmentagency 73 | organization-media/newspaper: newspaper 74 | organization-other: organization else 75 | organization-politicalparty: parliamentary 76 | organization-religion: faith 77 | organization-showorganization: showorganization 78 | organization-sportsleague: sportsleague 79 | organization-sportsteam: sportsteam 80 | other-astronomything: astronomything 81 | other-award: award 82 | other-biologything: biologic 83 | other-chemicalthing: chemistry 84 | other-currency: currency 85 | other-disease: disease 86 | other-educationaldegree: academic qualification 87 | other-god: belief 88 | other-language: language 89 | other-law: law 90 | other-livingthing: livingthing 91 | other-medical: medical 92 | person-actor: performer 93 | person-artist/author: author 94 | person-athlete: sportsman 95 | person-director: director 96 | person-other: person other 97 | person-politician: statesman 98 | person-scholar: scholar 99 | person-soldier: soldier 100 | product-airplane: aircraft 101 | product-car: autocar 102 | product-food: foodstuff 103 | product-game: game 104 | product-other: product otherness 105 | product-ship: ship 106 | product-software: program 107 | product-train: train 108 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V1.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | delete_list: 11 | mapper: 12 | O: O 13 | art-broadcastprogram: art 14 | art-film: art 15 | art-music: art 16 | art-other: art 17 | art-painting: art 18 | art-writtenart: art 19 | building-airport: building 20 | building-hospital: building 21 | building-hotel: building 22 | building-library: building 23 | building-other: building 24 | building-restaurant: building 25 | building-sportsfacility: building 26 | building-theater: theater 27 | event-attack/battle/war/militaryconflict: attack 28 | event-disaster: event 29 | event-election: event 30 | event-other: event other 31 | event-protest: event 32 | event-sportsevent: sportsevent 33 | location-GPE: geographical social political 34 | location-bodiesofwater: bodiesofwater 35 | location-island: location 36 | location-mountain: location 37 | location-other: location other 38 | location-park: park 39 | location-road/railway/highway/transit: location 40 | organization-company: organization 41 | organization-education: education 42 | organization-government/governmentagency: organization 43 | organization-media/newspaper: media 44 | organization-other: organization other 45 | organization-politicalparty: organization 46 | organization-religion: organization 47 | organization-showorganization: organization 48 | organization-sportsleague: organization 49 | organization-sportsteam: organization 50 | other-astronomything: astronomything 51 | other-award: award 52 | other-biologything: biologything 53 | other-chemicalthing: chemicalthing 54 | other-currency: currency 55 | other-disease: disease 56 | other-educationaldegree: educationaldegree 57 | other-god: god 58 | other-language: language 59 | other-law: law 60 | other-livingthing: livingthing 61 | other-medical: medical 62 | person-actor: person 63 | person-artist/author: person 64 | person-athlete: person 65 | person-director: person 66 | person-other: person 67 | person-politician: person 68 | person-scholar: person 69 | person-soldier: person 70 | product-airplane: airplane 71 | product-car: car 72 | product-food: food 73 | product-game: game 74 | product-other: product other 75 | product-ship: ship 76 | product-software: product 77 | product-train: product 78 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V2.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - disaster 11 | - election 12 | - library 13 | - person other 14 | - protest 15 | - train 16 | delete_list: 17 | mapper: 18 | O: O 19 | art-broadcastprogram: art 20 | art-film: art 21 | art-music: art 22 | art-other: art 23 | art-painting: art 24 | art-writtenart: art 25 | building-airport: building 26 | building-hospital: building 27 | building-hotel: building 28 | building-library: library 29 | building-other: building 30 | building-restaurant: building 31 | building-sportsfacility: building 32 | building-theater: theater 33 | event-attack/battle/war/militaryconflict: attack 34 | event-disaster: disaster 35 | event-election: election 36 | event-other: event other 37 | event-protest: protest 38 | event-sportsevent: sportsevent 39 | location-GPE: geographical social political 40 | location-bodiesofwater: bodiesofwater 41 | location-island: location 42 | location-mountain: location 43 | location-other: location other 44 | location-park: park 45 | location-road/railway/highway/transit: location 46 | organization-company: organization 47 | organization-education: education 48 | organization-government/governmentagency: organization 49 | organization-media/newspaper: media 50 | organization-other: organization other 51 | organization-politicalparty: organization 52 | organization-religion: organization 53 | organization-showorganization: organization 54 | organization-sportsleague: organization 55 | organization-sportsteam: organization 56 | other-astronomything: astronomything 57 | other-award: award 58 | other-biologything: biologything 59 | other-chemicalthing: chemicalthing 60 | other-currency: currency 61 | other-disease: disease 62 | other-educationaldegree: educationaldegree 63 | other-god: god 64 | other-language: language 65 | other-law: law 66 | other-livingthing: livingthing 67 | other-medical: medical 68 | person-actor: person 69 | person-artist/author: person 70 | person-athlete: person 71 | person-director: person 72 | person-other: person other 73 | person-politician: person 74 | person-scholar: person 75 | person-soldier: person 76 | product-airplane: airplane 77 | product-car: car 78 | product-food: food 79 | product-game: game 80 | product-other: product other 81 | product-ship: ship 82 | product-software: product 83 | product-train: train 84 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V3.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - athlete 12 | - building other 13 | - disaster 14 | - election 15 | - library 16 | - person other 17 | - protest 18 | - restaurant 19 | - road 20 | - sportsfacility 21 | - train 22 | delete_list: 23 | mapper: 24 | O: O 25 | art-broadcastprogram: art 26 | art-film: art 27 | art-music: art 28 | art-other: art 29 | art-painting: art 30 | art-writtenart: art 31 | building-airport: building 32 | building-hospital: building 33 | building-hotel: building 34 | building-library: library 35 | building-other: building other 36 | building-restaurant: restaurant 37 | building-sportsfacility: sportsfacility 38 | building-theater: theater 39 | event-attack/battle/war/militaryconflict: attack 40 | event-disaster: disaster 41 | event-election: election 42 | event-other: event other 43 | event-protest: protest 44 | event-sportsevent: sportsevent 45 | location-GPE: geographical social political 46 | location-bodiesofwater: bodiesofwater 47 | location-island: location 48 | location-mountain: location 49 | location-other: location other 50 | location-park: park 51 | location-road/railway/highway/transit: road 52 | organization-company: organization 53 | organization-education: education 54 | organization-government/governmentagency: organization 55 | organization-media/newspaper: media 56 | organization-other: organization other 57 | organization-politicalparty: organization 58 | organization-religion: organization 59 | organization-showorganization: organization 60 | organization-sportsleague: organization 61 | organization-sportsteam: organization 62 | other-astronomything: astronomything 63 | other-award: award 64 | other-biologything: biologything 65 | other-chemicalthing: chemicalthing 66 | other-currency: currency 67 | other-disease: disease 68 | other-educationaldegree: educationaldegree 69 | other-god: god 70 | other-language: language 71 | other-law: law 72 | other-livingthing: livingthing 73 | other-medical: medical 74 | person-actor: actor 75 | person-artist/author: person 76 | person-athlete: athlete 77 | person-director: person 78 | person-other: person other 79 | person-politician: person 80 | person-scholar: person 81 | person-soldier: person 82 | product-airplane: airplane 83 | product-car: car 84 | product-food: food 85 | product-game: game 86 | product-other: product other 87 | product-ship: ship 88 | product-software: product 89 | product-train: train 90 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V4.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - athlete 13 | - building other 14 | - disaster 15 | - election 16 | - hospital 17 | - library 18 | - person other 19 | - politician 20 | - protest 21 | - restaurant 22 | - road 23 | - scholar 24 | - soldier 25 | - sportsfacility 26 | - train 27 | - writtenart 28 | delete_list: 29 | mapper: 30 | O: O 31 | art-broadcastprogram: art 32 | art-film: art 33 | art-music: art 34 | art-other: art 35 | art-painting: art 36 | art-writtenart: writtenart 37 | building-airport: airport 38 | building-hospital: hospital 39 | building-hotel: building 40 | building-library: library 41 | building-other: building other 42 | building-restaurant: restaurant 43 | building-sportsfacility: sportsfacility 44 | building-theater: theater 45 | event-attack/battle/war/militaryconflict: attack 46 | event-disaster: disaster 47 | event-election: election 48 | event-other: event other 49 | event-protest: protest 50 | event-sportsevent: sportsevent 51 | location-GPE: geographical social political 52 | location-bodiesofwater: bodiesofwater 53 | location-island: location 54 | location-mountain: location 55 | location-other: location other 56 | location-park: park 57 | location-road/railway/highway/transit: road 58 | organization-company: organization 59 | organization-education: education 60 | organization-government/governmentagency: organization 61 | organization-media/newspaper: media 62 | organization-other: organization other 63 | organization-politicalparty: organization 64 | organization-religion: organization 65 | organization-showorganization: organization 66 | organization-sportsleague: organization 67 | organization-sportsteam: organization 68 | other-astronomything: astronomything 69 | other-award: award 70 | other-biologything: biologything 71 | other-chemicalthing: chemicalthing 72 | other-currency: currency 73 | other-disease: disease 74 | other-educationaldegree: educationaldegree 75 | other-god: god 76 | other-language: language 77 | other-law: law 78 | other-livingthing: livingthing 79 | other-medical: medical 80 | person-actor: actor 81 | person-artist/author: person 82 | person-athlete: athlete 83 | person-director: person 84 | person-other: person other 85 | person-politician: politician 86 | person-scholar: scholar 87 | person-soldier: soldier 88 | product-airplane: airplane 89 | product-car: car 90 | product-food: food 91 | product-game: game 92 | product-other: product other 93 | product-ship: ship 94 | product-software: product 95 | product-train: train 96 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V5.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art other 13 | - athlete 14 | - building other 15 | - disaster 16 | - election 17 | - government 18 | - hospital 19 | - library 20 | - music 21 | - painting 22 | - person other 23 | - politician 24 | - protest 25 | - restaurant 26 | - road 27 | - scholar 28 | - soldier 29 | - sportsfacility 30 | - sportsleague 31 | - sportsteam 32 | - train 33 | - writtenart 34 | delete_list: 35 | mapper: 36 | O: O 37 | art-broadcastprogram: art 38 | art-film: art 39 | art-music: music 40 | art-other: art other 41 | art-painting: painting 42 | art-writtenart: writtenart 43 | building-airport: airport 44 | building-hospital: hospital 45 | building-hotel: building 46 | building-library: library 47 | building-other: building other 48 | building-restaurant: restaurant 49 | building-sportsfacility: sportsfacility 50 | building-theater: theater 51 | event-attack/battle/war/militaryconflict: attack 52 | event-disaster: disaster 53 | event-election: election 54 | event-other: event other 55 | event-protest: protest 56 | event-sportsevent: sportsevent 57 | location-GPE: geographical social political 58 | location-bodiesofwater: bodiesofwater 59 | location-island: location 60 | location-mountain: location 61 | location-other: location other 62 | location-park: park 63 | location-road/railway/highway/transit: road 64 | organization-company: organization 65 | organization-education: education 66 | organization-government/governmentagency: government 67 | organization-media/newspaper: media 68 | organization-other: organization other 69 | organization-politicalparty: organization 70 | organization-religion: organization 71 | organization-showorganization: organization 72 | organization-sportsleague: sportsleague 73 | organization-sportsteam: sportsteam 74 | other-astronomything: astronomything 75 | other-award: award 76 | other-biologything: biologything 77 | other-chemicalthing: chemicalthing 78 | other-currency: currency 79 | other-disease: disease 80 | other-educationaldegree: educationaldegree 81 | other-god: god 82 | other-language: language 83 | other-law: law 84 | other-livingthing: livingthing 85 | other-medical: medical 86 | person-actor: actor 87 | person-artist/author: person 88 | person-athlete: athlete 89 | person-director: person 90 | person-other: person other 91 | person-politician: politician 92 | person-scholar: scholar 93 | person-soldier: soldier 94 | product-airplane: airplane 95 | product-car: car 96 | product-food: food 97 | product-game: game 98 | product-other: product other 99 | product-ship: ship 100 | product-software: product 101 | product-train: train 102 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V6.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art other 13 | - athlete 14 | - broadcastprogram 15 | - building other 16 | - company 17 | - disaster 18 | - election 19 | - film 20 | - government 21 | - hospital 22 | - hotel 23 | - island 24 | - library 25 | - mountain 26 | - music 27 | - painting 28 | - person other 29 | - politician 30 | - protest 31 | - restaurant 32 | - road 33 | - scholar 34 | - soldier 35 | - sportsfacility 36 | - sportsleague 37 | - sportsteam 38 | - train 39 | - writtenart 40 | delete_list: 41 | mapper: 42 | O: O 43 | art-broadcastprogram: broadcastprogram 44 | art-film: film 45 | art-music: music 46 | art-other: art other 47 | art-painting: painting 48 | art-writtenart: writtenart 49 | building-airport: airport 50 | building-hospital: hospital 51 | building-hotel: hotel 52 | building-library: library 53 | building-other: building other 54 | building-restaurant: restaurant 55 | building-sportsfacility: sportsfacility 56 | building-theater: theater 57 | event-attack/battle/war/militaryconflict: attack 58 | event-disaster: disaster 59 | event-election: election 60 | event-other: event other 61 | event-protest: protest 62 | event-sportsevent: sportsevent 63 | location-GPE: geographical social political 64 | location-bodiesofwater: bodiesofwater 65 | location-island: island 66 | location-mountain: mountain 67 | location-other: location other 68 | location-park: park 69 | location-road/railway/highway/transit: road 70 | organization-company: company 71 | organization-education: education 72 | organization-government/governmentagency: government 73 | organization-media/newspaper: media 74 | organization-other: organization other 75 | organization-politicalparty: organization 76 | organization-religion: organization 77 | organization-showorganization: organization 78 | organization-sportsleague: sportsleague 79 | organization-sportsteam: sportsteam 80 | other-astronomything: astronomything 81 | other-award: award 82 | other-biologything: biologything 83 | other-chemicalthing: chemicalthing 84 | other-currency: currency 85 | other-disease: disease 86 | other-educationaldegree: educationaldegree 87 | other-god: god 88 | other-language: language 89 | other-law: law 90 | other-livingthing: livingthing 91 | other-medical: medical 92 | person-actor: actor 93 | person-artist/author: person 94 | person-athlete: athlete 95 | person-director: person 96 | person-other: person other 97 | person-politician: politician 98 | person-scholar: scholar 99 | person-soldier: soldier 100 | product-airplane: airplane 101 | product-car: car 102 | product-food: food 103 | product-game: game 104 | product-other: product other 105 | product-ship: ship 106 | product-software: product 107 | product-train: train 108 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/entity/Few-NERD_V7.yaml: -------------------------------------------------------------------------------- 1 | name: Few-NERD 2 | path: data/Few-NERD 3 | data_class: NERD 4 | split: 5 | train: train.txt 6 | val: dev.txt 7 | test: test.txt 8 | language: en 9 | new_list: 10 | - actor 11 | - airport 12 | - art other 13 | - artist 14 | - athlete 15 | - broadcastprogram 16 | - building other 17 | - company 18 | - director 19 | - disaster 20 | - election 21 | - film 22 | - government 23 | - hospital 24 | - hotel 25 | - island 26 | - library 27 | - mountain 28 | - music 29 | - painting 30 | - person other 31 | - politicalparty 32 | - politician 33 | - protest 34 | - religion 35 | - restaurant 36 | - road 37 | - scholar 38 | - showorganization 39 | - software 40 | - soldier 41 | - sportsfacility 42 | - sportsleague 43 | - sportsteam 44 | - train 45 | - writtenart 46 | delete_list: 47 | mapper: 48 | O: O 49 | art-broadcastprogram: broadcastprogram 50 | art-film: film 51 | art-music: music 52 | art-other: art other 53 | art-painting: painting 54 | art-writtenart: writtenart 55 | building-airport: airport 56 | building-hospital: hospital 57 | building-hotel: hotel 58 | building-library: library 59 | building-other: building other 60 | building-restaurant: restaurant 61 | building-sportsfacility: sportsfacility 62 | building-theater: theater 63 | event-attack/battle/war/militaryconflict: attack 64 | event-disaster: disaster 65 | event-election: election 66 | event-other: event other 67 | event-protest: protest 68 | event-sportsevent: sportsevent 69 | location-GPE: geographical social political 70 | location-bodiesofwater: bodiesofwater 71 | location-island: island 72 | location-mountain: mountain 73 | location-other: location other 74 | location-park: park 75 | location-road/railway/highway/transit: road 76 | organization-company: company 77 | organization-education: education 78 | organization-government/governmentagency: government 79 | organization-media/newspaper: media 80 | organization-other: organization other 81 | organization-politicalparty: politicalparty 82 | organization-religion: religion 83 | organization-showorganization: showorganization 84 | organization-sportsleague: sportsleague 85 | organization-sportsteam: sportsteam 86 | other-astronomything: astronomything 87 | other-award: award 88 | other-biologything: biologything 89 | other-chemicalthing: chemicalthing 90 | other-currency: currency 91 | other-disease: disease 92 | other-educationaldegree: educationaldegree 93 | other-god: god 94 | other-language: language 95 | other-law: law 96 | other-livingthing: livingthing 97 | other-medical: medical 98 | person-actor: actor 99 | person-artist/author: artist 100 | person-athlete: athlete 101 | person-director: director 102 | person-other: person other 103 | person-politician: politician 104 | person-scholar: scholar 105 | person-soldier: soldier 106 | product-airplane: airplane 107 | product-car: car 108 | product-food: food 109 | product-game: game 110 | product-other: product other 111 | product-ship: ship 112 | product-software: software 113 | product-train: train 114 | product-weapon: weapon -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H1.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | delete_list: 11 | - Contact:Meet 12 | - Life:Divorce 13 | - Personnel:Nominate 14 | - Business:Merge-Org 15 | - Justice:Charge-Indict 16 | - Justice:Release-Parole 17 | - Business:Declare-Bankruptcy 18 | - Justice:Appeal 19 | - Justice:Trial-Hearing 20 | - Business:Start-Org 21 | - Justice:Pardon 22 | - Personnel:End-Position 23 | - Conflict:Demonstrate 24 | - Justice:Execute 25 | - Transaction:Transfer-Money 26 | - Justice:Fine 27 | - Life:Injure 28 | - Justice:Convict 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place 98 | 99 | 100 | 101 | 102 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H2.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - convict 11 | - fine 12 | - injure 13 | delete_list: 14 | - Contact:Meet 15 | - Life:Divorce 16 | - Personnel:Nominate 17 | - Business:Merge-Org 18 | - Justice:Charge-Indict 19 | - Justice:Release-Parole 20 | - Business:Declare-Bankruptcy 21 | - Justice:Appeal 22 | - Justice:Trial-Hearing 23 | - Business:Start-Org 24 | - Justice:Pardon 25 | - Personnel:End-Position 26 | - Conflict:Demonstrate 27 | - Justice:Execute 28 | - Transaction:Transfer-Money 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place 98 | 99 | 100 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H3.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - convict 11 | - demonstrate 12 | - execute 13 | - fine 14 | - injure 15 | - transfer money 16 | delete_list: 17 | - Contact:Meet 18 | - Life:Divorce 19 | - Personnel:Nominate 20 | - Business:Merge-Org 21 | - Justice:Charge-Indict 22 | - Justice:Release-Parole 23 | - Business:Declare-Bankruptcy 24 | - Justice:Appeal 25 | - Justice:Trial-Hearing 26 | - Business:Start-Org 27 | - Justice:Pardon 28 | - Personnel:End-Position 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place 98 | 99 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H4.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - convict 11 | - demonstrate 12 | - end position 13 | - execute 14 | - fine 15 | - injure 16 | - pardon 17 | - start organization 18 | - transfer money 19 | delete_list: 20 | - Contact:Meet 21 | - Life:Divorce 22 | - Personnel:Nominate 23 | - Business:Merge-Org 24 | - Justice:Charge-Indict 25 | - Justice:Release-Parole 26 | - Business:Declare-Bankruptcy 27 | - Justice:Appeal 28 | - Justice:Trial-Hearing 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H5.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - appeal 11 | - convict 12 | - declare bankruptcy 13 | - demonstrate 14 | - end position 15 | - execute 16 | - fine 17 | - injure 18 | - pardon 19 | - start organization 20 | - transfer money 21 | - trial hearing 22 | delete_list: 23 | - Contact:Meet 24 | - Life:Divorce 25 | - Personnel:Nominate 26 | - Business:Merge-Org 27 | - Justice:Charge-Indict 28 | - Justice:Release-Parole 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H6.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - appeal 11 | - charge indict 12 | - convict 13 | - declare bankruptcy 14 | - demonstrate 15 | - end position 16 | - execute 17 | - fine 18 | - injure 19 | - merge organization 20 | - pardon 21 | - release parole 22 | - start organization 23 | - transfer money 24 | - trial hearing 25 | delete_list: 26 | - Contact:Meet 27 | - Life:Divorce 28 | - Personnel:Nominate 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_H7.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - appeal 11 | - charge indict 12 | - convict 13 | - declare bankruptcy 14 | - demonstrate 15 | - divorce 16 | - end position 17 | - execute 18 | - fine 19 | - injure 20 | - meet 21 | - merge organization 22 | - nominate 23 | - pardon 24 | - release parole 25 | - start organization 26 | - transfer money 27 | - trial hearing 28 | delete_list: 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M1.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | - Business:Declare-Bankruptcy 11 | - Business:End-Org 12 | - Business:Merge-Org 13 | - Business:Start-Org 14 | - Contact:Meet 15 | - Contact:Phone-Write 16 | - Personnel:Elect 17 | - Personnel:End-Position 18 | - Personnel:Nominate 19 | - Personnel:Start-Position 20 | new_list: 21 | mapper: 22 | Business:Declare-Bankruptcy: business 23 | Business:End-Org: business 24 | Business:Merge-Org: business 25 | Business:Start-Org: business 26 | Conflict:Attack: attack 27 | Conflict:Demonstrate: demonstrate 28 | Contact:Meet: meet 29 | Contact:Phone-Write: phone write 30 | Justice:Acquit: justice 31 | Justice:Appeal: justice 32 | Justice:Arrest-Jail: justice 33 | Justice:Charge-Indict: justice 34 | Justice:Convict: justice 35 | Justice:Execute: execute 36 | Justice:Extradite: extradite 37 | Justice:Fine: fine 38 | Justice:Pardon: pardon 39 | Justice:Release-Parole: release parole 40 | Justice:Sentence: sentence 41 | Justice:Sue: sue 42 | Justice:Trial-Hearing: justice 43 | Life:Be-Born: be born 44 | Life:Die: die 45 | Life:Divorce: life 46 | Life:Injure: life 47 | Life:Marry: marry 48 | Movement:Transport: transport 49 | Personnel:Elect: elect 50 | Personnel:End-Position: end position 51 | Personnel:Nominate: nominate 52 | Personnel:Start-Position: start position 53 | Transaction:Transfer-Money: transfer money 54 | Transaction:Transfer-Ownership: transfer ownership 55 | FAC: facility 56 | GPE: geographical social political 57 | LOC: location 58 | ORG: organization 59 | PER: person 60 | VEH: vehicle 61 | WEA: weapon 62 | ORG-AFF: organization affiliation 63 | GEN-AFF: general affiliation 64 | PHYS: physical 65 | PART-WHOLE: part whole 66 | PER-SOC: personal social 67 | ART: agent artifact 68 | Defendant: defendant 69 | Prosecutor: prosecutor 70 | Person: person 71 | Origin: origin 72 | Buyer: buyer 73 | Plaintiff: plaintiff 74 | Victim: victim 75 | Org: organization 76 | Adjudicator: adjudicator 77 | Seller: seller 78 | Beneficiary: beneficiary 79 | Giver: giver 80 | Target: target 81 | Agent: agent 82 | Instrument: instrument 83 | Vehicle: vehicle 84 | Entity: entity 85 | Destination: destination 86 | Recipient: recipient 87 | Attacker: attacker 88 | Artifact: artifact 89 | Place: place 90 | 91 | 92 | 93 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M2.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - acquit 11 | - appeal 12 | - contact 13 | - meet 14 | delete_list: 15 | - Business:Declare-Bankruptcy 16 | - Business:End-Org 17 | - Business:Merge-Org 18 | - Business:Start-Org 19 | - Personnel:Elect 20 | - Personnel:End-Position 21 | - Personnel:Nominate 22 | - Personnel:Start-Position 23 | mapper: 24 | Business:Declare-Bankruptcy: business 25 | Business:End-Org: business 26 | Business:Merge-Org: business 27 | Business:Start-Org: business 28 | Conflict:Attack: attack 29 | Conflict:Demonstrate: demonstrate 30 | Contact:Meet: meet 31 | Contact:Phone-Write: contact 32 | Justice:Acquit: acquit 33 | Justice:Appeal: appeal 34 | Justice:Arrest-Jail: justice 35 | Justice:Charge-Indict: justice 36 | Justice:Convict: justice 37 | Justice:Execute: execute 38 | Justice:Extradite: extradite 39 | Justice:Fine: fine 40 | Justice:Pardon: pardon 41 | Justice:Release-Parole: release parole 42 | Justice:Sentence: sentence 43 | Justice:Sue: sue 44 | Justice:Trial-Hearing: justice 45 | Life:Be-Born: be born 46 | Life:Die: die 47 | Life:Divorce: life 48 | Life:Injure: life 49 | Life:Marry: marry 50 | Movement:Transport: transport 51 | Personnel:Elect: elect 52 | Personnel:End-Position: end position 53 | Personnel:Nominate: nominate 54 | Personnel:Start-Position: start position 55 | Transaction:Transfer-Money: transfer money 56 | Transaction:Transfer-Ownership: transfer ownership 57 | FAC: facility 58 | GPE: geographical social political 59 | LOC: location 60 | ORG: organization 61 | PER: person 62 | VEH: vehicle 63 | WEA: weapon 64 | ORG-AFF: organization affiliation 65 | GEN-AFF: general affiliation 66 | PHYS: physical 67 | PART-WHOLE: part whole 68 | PER-SOC: personal social 69 | ART: agent artifact 70 | Defendant: defendant 71 | Prosecutor: prosecutor 72 | Person: person 73 | Origin: origin 74 | Buyer: buyer 75 | Plaintiff: plaintiff 76 | Victim: victim 77 | Org: organization 78 | Adjudicator: adjudicator 79 | Seller: seller 80 | Beneficiary: beneficiary 81 | Giver: giver 82 | Target: target 83 | Agent: agent 84 | Instrument: instrument 85 | Vehicle: vehicle 86 | Entity: entity 87 | Destination: destination 88 | Recipient: recipient 89 | Attacker: attacker 90 | Artifact: artifact 91 | Place: place 92 | 93 | 94 | 95 | 96 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M3.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - acquit 11 | - appeal 12 | - business 13 | - contact 14 | - meet 15 | - merge organization 16 | - start organization 17 | - trial hearing 18 | delete_list: 19 | - Personnel:Elect 20 | - Personnel:End-Position 21 | - Personnel:Nominate 22 | - Personnel:Start-Position 23 | mapper: 24 | Business:Declare-Bankruptcy: business 25 | Business:End-Org: business 26 | Business:Merge-Org: merge organization 27 | Business:Start-Org: start organization 28 | Conflict:Attack: attack 29 | Conflict:Demonstrate: demonstrate 30 | Contact:Meet: meet 31 | Contact:Phone-Write: contact 32 | Justice:Acquit: acquit 33 | Justice:Appeal: appeal 34 | Justice:Arrest-Jail: justice 35 | Justice:Charge-Indict: justice 36 | Justice:Convict: justice 37 | Justice:Execute: execute 38 | Justice:Extradite: extradite 39 | Justice:Fine: fine 40 | Justice:Pardon: pardon 41 | Justice:Release-Parole: release parole 42 | Justice:Sentence: sentence 43 | Justice:Sue: sue 44 | Justice:Trial-Hearing: trial hearing 45 | Life:Be-Born: be born 46 | Life:Die: die 47 | Life:Divorce: life 48 | Life:Injure: life 49 | Life:Marry: marry 50 | Movement:Transport: transport 51 | Personnel:Elect: elect 52 | Personnel:End-Position: end position 53 | Personnel:Nominate: nominate 54 | Personnel:Start-Position: start position 55 | Transaction:Transfer-Money: transfer money 56 | Transaction:Transfer-Ownership: transfer ownership 57 | FAC: facility 58 | GPE: geographical social political 59 | LOC: location 60 | ORG: organization 61 | PER: person 62 | VEH: vehicle 63 | WEA: weapon 64 | ORG-AFF: organization affiliation 65 | GEN-AFF: general affiliation 66 | PHYS: physical 67 | PART-WHOLE: part whole 68 | PER-SOC: personal social 69 | ART: agent artifact 70 | Defendant: defendant 71 | Prosecutor: prosecutor 72 | Person: person 73 | Origin: origin 74 | Buyer: buyer 75 | Plaintiff: plaintiff 76 | Victim: victim 77 | Org: organization 78 | Adjudicator: adjudicator 79 | Seller: seller 80 | Beneficiary: beneficiary 81 | Giver: giver 82 | Target: target 83 | Agent: agent 84 | Instrument: instrument 85 | Vehicle: vehicle 86 | Entity: entity 87 | Destination: destination 88 | Recipient: recipient 89 | Attacker: attacker 90 | Artifact: artifact 91 | Place: place 92 | 93 | 94 | 95 | 96 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M4.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - acquit 11 | - appeal 12 | - declare bankruptcy 13 | - end organization 14 | - meet 15 | - merge organization 16 | - phone write 17 | - start organization 18 | - trial hearing 19 | delete_list: 20 | - Personnel:Elect 21 | - Personnel:End-Position 22 | - Personnel:Nominate 23 | - Personnel:Start-Position 24 | mapper: 25 | Business:Declare-Bankruptcy: declare bankruptcy 26 | Business:End-Org: end organization 27 | Business:Merge-Org: merge organization 28 | Business:Start-Org: start organization 29 | Conflict:Attack: attack 30 | Conflict:Demonstrate: demonstrate 31 | Contact:Meet: meet 32 | Contact:Phone-Write: phone write 33 | Justice:Acquit: acquit 34 | Justice:Appeal: appeal 35 | Justice:Arrest-Jail: justice 36 | Justice:Charge-Indict: justice 37 | Justice:Convict: justice 38 | Justice:Execute: execute 39 | Justice:Extradite: extradite 40 | Justice:Fine: fine 41 | Justice:Pardon: pardon 42 | Justice:Release-Parole: release parole 43 | Justice:Sentence: sentence 44 | Justice:Sue: sue 45 | Justice:Trial-Hearing: trial hearing 46 | Life:Be-Born: be born 47 | Life:Die: die 48 | Life:Divorce: life 49 | Life:Injure: life 50 | Life:Marry: marry 51 | Movement:Transport: transport 52 | Personnel:Elect: elect 53 | Personnel:End-Position: end position 54 | Personnel:Nominate: nominate 55 | Personnel:Start-Position: start position 56 | Transaction:Transfer-Money: transfer money 57 | Transaction:Transfer-Ownership: transfer ownership 58 | FAC: facility 59 | GPE: geographical social political 60 | LOC: location 61 | ORG: organization 62 | PER: person 63 | VEH: vehicle 64 | WEA: weapon 65 | ORG-AFF: organization affiliation 66 | GEN-AFF: general affiliation 67 | PHYS: physical 68 | PART-WHOLE: part whole 69 | PER-SOC: personal social 70 | ART: agent artifact 71 | Defendant: defendant 72 | Prosecutor: prosecutor 73 | Person: person 74 | Origin: origin 75 | Buyer: buyer 76 | Plaintiff: plaintiff 77 | Victim: victim 78 | Org: organization 79 | Adjudicator: adjudicator 80 | Seller: seller 81 | Beneficiary: beneficiary 82 | Giver: giver 83 | Target: target 84 | Agent: agent 85 | Instrument: instrument 86 | Vehicle: vehicle 87 | Entity: entity 88 | Destination: destination 89 | Recipient: recipient 90 | Attacker: attacker 91 | Artifact: artifact 92 | Place: place 93 | 94 | 95 | 96 | 97 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M5.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - acquit 11 | - appeal 12 | - arrest jail 13 | - charge indict 14 | - convict 15 | - declare bankruptcy 16 | - end organization 17 | - meet 18 | - merge organization 19 | - phone write 20 | - start organization 21 | - trial hearing 22 | delete_list: 23 | - Personnel:Elect 24 | - Personnel:End-Position 25 | - Personnel:Nominate 26 | - Personnel:Start-Position 27 | mapper: 28 | Business:Declare-Bankruptcy: declare bankruptcy 29 | Business:End-Org: end organization 30 | Business:Merge-Org: merge organization 31 | Business:Start-Org: start organization 32 | Conflict:Attack: attack 33 | Conflict:Demonstrate: demonstrate 34 | Contact:Meet: meet 35 | Contact:Phone-Write: phone write 36 | Justice:Acquit: acquit 37 | Justice:Appeal: appeal 38 | Justice:Arrest-Jail: arrest jail 39 | Justice:Charge-Indict: charge indict 40 | Justice:Convict: convict 41 | Justice:Execute: execute 42 | Justice:Extradite: extradite 43 | Justice:Fine: fine 44 | Justice:Pardon: pardon 45 | Justice:Release-Parole: release parole 46 | Justice:Sentence: sentence 47 | Justice:Sue: sue 48 | Justice:Trial-Hearing: trial hearing 49 | Life:Be-Born: be born 50 | Life:Die: die 51 | Life:Divorce: life 52 | Life:Injure: life 53 | Life:Marry: marry 54 | Movement:Transport: transport 55 | Personnel:Elect: elect 56 | Personnel:End-Position: end position 57 | Personnel:Nominate: nominate 58 | Personnel:Start-Position: start position 59 | Transaction:Transfer-Money: transfer money 60 | Transaction:Transfer-Ownership: transfer ownership 61 | FAC: facility 62 | GPE: geographical social political 63 | LOC: location 64 | ORG: organization 65 | PER: person 66 | VEH: vehicle 67 | WEA: weapon 68 | ORG-AFF: organization affiliation 69 | GEN-AFF: general affiliation 70 | PHYS: physical 71 | PART-WHOLE: part whole 72 | PER-SOC: personal social 73 | ART: agent artifact 74 | Defendant: defendant 75 | Prosecutor: prosecutor 76 | Person: person 77 | Origin: origin 78 | Buyer: buyer 79 | Plaintiff: plaintiff 80 | Victim: victim 81 | Org: organization 82 | Adjudicator: adjudicator 83 | Seller: seller 84 | Beneficiary: beneficiary 85 | Giver: giver 86 | Target: target 87 | Agent: agent 88 | Instrument: instrument 89 | Vehicle: vehicle 90 | Entity: entity 91 | Destination: destination 92 | Recipient: recipient 93 | Attacker: attacker 94 | Artifact: artifact 95 | Place: place 96 | 97 | 98 | 99 | 100 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M6.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - acquit 11 | - appeal 12 | - arrest jail 13 | - charge indict 14 | - convict 15 | - declare bankruptcy 16 | - divorce 17 | - end organization 18 | - end position 19 | - injure 20 | - meet 21 | - merge organization 22 | - personnel 23 | - phone write 24 | - start organization 25 | - trial hearing 26 | delete_list: 27 | mapper: 28 | Business:Declare-Bankruptcy: declare bankruptcy 29 | Business:End-Org: end organization 30 | Business:Merge-Org: merge organization 31 | Business:Start-Org: start organization 32 | Conflict:Attack: attack 33 | Conflict:Demonstrate: demonstrate 34 | Contact:Meet: meet 35 | Contact:Phone-Write: phone write 36 | Justice:Acquit: acquit 37 | Justice:Appeal: appeal 38 | Justice:Arrest-Jail: arrest jail 39 | Justice:Charge-Indict: charge indict 40 | Justice:Convict: convict 41 | Justice:Execute: execute 42 | Justice:Extradite: extradite 43 | Justice:Fine: fine 44 | Justice:Pardon: pardon 45 | Justice:Release-Parole: release parole 46 | Justice:Sentence: sentence 47 | Justice:Sue: sue 48 | Justice:Trial-Hearing: trial hearing 49 | Life:Be-Born: be born 50 | Life:Die: die 51 | Life:Divorce: divorce 52 | Life:Injure: injure 53 | Life:Marry: marry 54 | Movement:Transport: transport 55 | Personnel:Elect: personnel 56 | Personnel:End-Position: end position 57 | Personnel:Nominate: personnel 58 | Personnel:Start-Position: personnel 59 | Transaction:Transfer-Money: transfer money 60 | Transaction:Transfer-Ownership: transfer ownership 61 | FAC: facility 62 | GPE: geographical social political 63 | LOC: location 64 | ORG: organization 65 | PER: person 66 | VEH: vehicle 67 | WEA: weapon 68 | ORG-AFF: organization affiliation 69 | GEN-AFF: general affiliation 70 | PHYS: physical 71 | PART-WHOLE: part whole 72 | PER-SOC: personal social 73 | ART: agent artifact 74 | Defendant: defendant 75 | Prosecutor: prosecutor 76 | Person: person 77 | Origin: origin 78 | Buyer: buyer 79 | Plaintiff: plaintiff 80 | Victim: victim 81 | Org: organization 82 | Adjudicator: adjudicator 83 | Seller: seller 84 | Beneficiary: beneficiary 85 | Giver: giver 86 | Target: target 87 | Agent: agent 88 | Instrument: instrument 89 | Vehicle: vehicle 90 | Entity: entity 91 | Destination: destination 92 | Recipient: recipient 93 | Attacker: attacker 94 | Artifact: artifact 95 | Place: place 96 | 97 | 98 | 99 | 100 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_M7.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - acquit 11 | - appeal 12 | - arrest jail 13 | - charge indict 14 | - convict 15 | - declare bankruptcy 16 | - divorce 17 | - elect 18 | - end organization 19 | - end position 20 | - injure 21 | - meet 22 | - merge organization 23 | - nominate 24 | - phone write 25 | - start organization 26 | - start position 27 | - trial hearing 28 | delete_list: 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place 98 | 99 | 100 | 101 | 102 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R1.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | delete_list: 11 | mapper: 12 | Business:Declare-Bankruptcy: declare bankruptcy 13 | Business:End-Org: end organization 14 | Business:Merge-Org: merge organization 15 | Business:Start-Org: start organization 16 | Conflict:Attack: attack 17 | Conflict:Demonstrate: demonstrate 18 | Contact:Meet: meet 19 | Contact:Phone-Write: phone write 20 | Justice:Acquit: acquit 21 | Justice:Appeal: appeal 22 | Justice:Arrest-Jail: arrest jail 23 | Justice:Charge-Indict: charge indict 24 | Justice:Convict: convict 25 | Justice:Execute: execute 26 | Justice:Extradite: extradite 27 | Justice:Fine: fine 28 | Justice:Pardon: pardon 29 | Justice:Release-Parole: release parole 30 | Justice:Sentence: sentence 31 | Justice:Sue: sue 32 | Justice:Trial-Hearing: trial hearing 33 | Life:Be-Born: be born 34 | Life:Die: die 35 | Life:Divorce: divorce 36 | Life:Injure: injure 37 | Life:Marry: marry 38 | Movement:Transport: transport 39 | Personnel:Elect: elect 40 | Personnel:End-Position: end position 41 | Personnel:Nominate: nominate 42 | Personnel:Start-Position: start position 43 | Transaction:Transfer-Money: transfer money 44 | Transaction:Transfer-Ownership: transfer ownership 45 | FAC: facility 46 | GPE: geographical social political 47 | LOC: location 48 | ORG: organization 49 | PER: person 50 | VEH: vehicle 51 | WEA: weapon 52 | ORG-AFF: organization affiliation 53 | GEN-AFF: general affiliation 54 | PHYS: physical 55 | PART-WHOLE: part whole 56 | PER-SOC: personal social 57 | ART: agent artifact 58 | Defendant: defendant 59 | Prosecutor: prosecutor 60 | Person: person 61 | Origin: origin 62 | Buyer: buyer 63 | Plaintiff: plaintiff 64 | Victim: victim 65 | Org: organization 66 | Adjudicator: adjudicator 67 | Seller: seller 68 | Beneficiary: beneficiary 69 | Giver: giver 70 | Target: target 71 | Agent: agent 72 | Instrument: instrument 73 | Vehicle: vehicle 74 | Entity: entity 75 | Destination: destination 76 | Recipient: recipient 77 | Attacker: attacker 78 | Artifact: artifact 79 | Place: place 80 | 81 | 82 | 83 | 84 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R2.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - carry 11 | - begin 12 | - separate 13 | delete_list: 14 | mapper: 15 | Business:Declare-Bankruptcy: declare bankruptcy 16 | Business:End-Org: end organization 17 | Business:Merge-Org: merge organization 18 | Business:Start-Org: start organization 19 | Conflict:Attack: attack 20 | Conflict:Demonstrate: demonstrate 21 | Contact:Meet: meet 22 | Contact:Phone-Write: phone write 23 | Justice:Acquit: acquit 24 | Justice:Appeal: appeal 25 | Justice:Arrest-Jail: arrest jail 26 | Justice:Charge-Indict: charge indict 27 | Justice:Convict: convict 28 | Justice:Execute: execute 29 | Justice:Extradite: extradite 30 | Justice:Fine: fine 31 | Justice:Pardon: pardon 32 | Justice:Release-Parole: release parole 33 | Justice:Sentence: sentence 34 | Justice:Sue: sue 35 | Justice:Trial-Hearing: trial hearing 36 | Life:Be-Born: be born 37 | Life:Die: die 38 | Life:Divorce: separate 39 | Life:Injure: injure 40 | Life:Marry: marry 41 | Movement:Transport: carry 42 | Personnel:Elect: elect 43 | Personnel:End-Position: end position 44 | Personnel:Nominate: nominate 45 | Personnel:Start-Position: begin 46 | Transaction:Transfer-Money: transfer money 47 | Transaction:Transfer-Ownership: transfer ownership 48 | FAC: facility 49 | GPE: geographical social political 50 | LOC: location 51 | ORG: organization 52 | PER: person 53 | VEH: vehicle 54 | WEA: weapon 55 | ORG-AFF: organization affiliation 56 | GEN-AFF: general affiliation 57 | PHYS: physical 58 | PART-WHOLE: part whole 59 | PER-SOC: personal social 60 | ART: agent artifact 61 | Defendant: defendant 62 | Prosecutor: prosecutor 63 | Person: person 64 | Origin: origin 65 | Buyer: buyer 66 | Plaintiff: plaintiff 67 | Victim: victim 68 | Org: organization 69 | Adjudicator: adjudicator 70 | Seller: seller 71 | Beneficiary: beneficiary 72 | Giver: giver 73 | Target: target 74 | Agent: agent 75 | Instrument: instrument 76 | Vehicle: vehicle 77 | Entity: entity 78 | Destination: destination 79 | Recipient: recipient 80 | Attacker: attacker 81 | Artifact: artifact 82 | Place: place 83 | 84 | 85 | 86 | 87 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R3.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - end 11 | - carry 12 | - separate 13 | - remittance 14 | - begin 15 | - prosecute 16 | delete_list: 17 | mapper: 18 | Business:Declare-Bankruptcy: declare bankruptcy 19 | Business:End-Org: end organization 20 | Business:Merge-Org: merge organization 21 | Business:Start-Org: start organization 22 | Conflict:Attack: attack 23 | Conflict:Demonstrate: demonstrate 24 | Contact:Meet: meet 25 | Contact:Phone-Write: phone write 26 | Justice:Acquit: acquit 27 | Justice:Appeal: appeal 28 | Justice:Arrest-Jail: arrest jail 29 | Justice:Charge-Indict: prosecute 30 | Justice:Convict: convict 31 | Justice:Execute: execute 32 | Justice:Extradite: extradite 33 | Justice:Fine: fine 34 | Justice:Pardon: pardon 35 | Justice:Release-Parole: release parole 36 | Justice:Sentence: sentence 37 | Justice:Sue: sue 38 | Justice:Trial-Hearing: trial hearing 39 | Life:Be-Born: be born 40 | Life:Die: die 41 | Life:Divorce: separate 42 | Life:Injure: injure 43 | Life:Marry: marry 44 | Movement:Transport: carry 45 | Personnel:Elect: elect 46 | Personnel:End-Position: end 47 | Personnel:Nominate: nominate 48 | Personnel:Start-Position: begin 49 | Transaction:Transfer-Money: remittance 50 | Transaction:Transfer-Ownership: transfer ownership 51 | FAC: facility 52 | GPE: geographical social political 53 | LOC: location 54 | ORG: organization 55 | PER: person 56 | VEH: vehicle 57 | WEA: weapon 58 | ORG-AFF: organization affiliation 59 | GEN-AFF: general affiliation 60 | PHYS: physical 61 | PART-WHOLE: part whole 62 | PER-SOC: personal social 63 | ART: agent artifact 64 | Defendant: defendant 65 | Prosecutor: prosecutor 66 | Person: person 67 | Origin: origin 68 | Buyer: buyer 69 | Plaintiff: plaintiff 70 | Victim: victim 71 | Org: organization 72 | Adjudicator: adjudicator 73 | Seller: seller 74 | Beneficiary: beneficiary 75 | Giver: giver 76 | Target: target 77 | Agent: agent 78 | Instrument: instrument 79 | Vehicle: vehicle 80 | Entity: entity 81 | Destination: destination 82 | Recipient: recipient 83 | Attacker: attacker 84 | Artifact: artifact 85 | Place: place 86 | 87 | 88 | 89 | 90 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R4.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - pass away 11 | - end 12 | - carry 13 | - separate 14 | - attend the trial 15 | - perform 16 | - remittance 17 | - begin 18 | - prosecute 19 | delete_list: 20 | mapper: 21 | Business:Declare-Bankruptcy: declare bankruptcy 22 | Business:End-Org: end organization 23 | Business:Merge-Org: merge organization 24 | Business:Start-Org: start organization 25 | Conflict:Attack: attack 26 | Conflict:Demonstrate: demonstrate 27 | Contact:Meet: meet 28 | Contact:Phone-Write: phone write 29 | Justice:Acquit: acquit 30 | Justice:Appeal: appeal 31 | Justice:Arrest-Jail: arrest jail 32 | Justice:Charge-Indict: prosecute 33 | Justice:Convict: convict 34 | Justice:Execute: perform 35 | Justice:Extradite: extradite 36 | Justice:Fine: fine 37 | Justice:Pardon: pardon 38 | Justice:Release-Parole: release parole 39 | Justice:Sentence: sentence 40 | Justice:Sue: sue 41 | Justice:Trial-Hearing: attend the trial 42 | Life:Be-Born: be born 43 | Life:Die: pass away 44 | Life:Divorce: separate 45 | Life:Injure: injure 46 | Life:Marry: marry 47 | Movement:Transport: carry 48 | Personnel:Elect: elect 49 | Personnel:End-Position: end 50 | Personnel:Nominate: nominate 51 | Personnel:Start-Position: begin 52 | Transaction:Transfer-Money: remittance 53 | Transaction:Transfer-Ownership: transfer ownership 54 | FAC: facility 55 | GPE: geographical social political 56 | LOC: location 57 | ORG: organization 58 | PER: person 59 | VEH: vehicle 60 | WEA: weapon 61 | ORG-AFF: organization affiliation 62 | GEN-AFF: general affiliation 63 | PHYS: physical 64 | PART-WHOLE: part whole 65 | PER-SOC: personal social 66 | ART: agent artifact 67 | Defendant: defendant 68 | Prosecutor: prosecutor 69 | Person: person 70 | Origin: origin 71 | Buyer: buyer 72 | Plaintiff: plaintiff 73 | Victim: victim 74 | Org: organization 75 | Adjudicator: adjudicator 76 | Seller: seller 77 | Beneficiary: beneficiary 78 | Giver: giver 79 | Target: target 80 | Agent: agent 81 | Instrument: instrument 82 | Vehicle: vehicle 83 | Entity: entity 84 | Destination: destination 85 | Recipient: recipient 86 | Attacker: attacker 87 | Artifact: artifact 88 | Place: place 89 | 90 | 91 | 92 | 93 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R5.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - condemn 11 | - pass away 12 | - go out of business 13 | - end 14 | - encounter 15 | - carry 16 | - separate 17 | - attend the trial 18 | - perform 19 | - remittance 20 | - begin 21 | - prosecute 22 | delete_list: 23 | mapper: 24 | Business:Declare-Bankruptcy: go out of business 25 | Business:End-Org: end organization 26 | Business:Merge-Org: merge organization 27 | Business:Start-Org: start organization 28 | Conflict:Attack: attack 29 | Conflict:Demonstrate: demonstrate 30 | Contact:Meet: encounter 31 | Contact:Phone-Write: phone write 32 | Justice:Acquit: acquit 33 | Justice:Appeal: appeal 34 | Justice:Arrest-Jail: arrest jail 35 | Justice:Charge-Indict: prosecute 36 | Justice:Convict: convict 37 | Justice:Execute: perform 38 | Justice:Extradite: extradite 39 | Justice:Fine: fine 40 | Justice:Pardon: pardon 41 | Justice:Release-Parole: release parole 42 | Justice:Sentence: condemn 43 | Justice:Sue: sue 44 | Justice:Trial-Hearing: attend the trial 45 | Life:Be-Born: be born 46 | Life:Die: pass away 47 | Life:Divorce: separate 48 | Life:Injure: injure 49 | Life:Marry: marry 50 | Movement:Transport: carry 51 | Personnel:Elect: elect 52 | Personnel:End-Position: end 53 | Personnel:Nominate: nominate 54 | Personnel:Start-Position: begin 55 | Transaction:Transfer-Money: remittance 56 | Transaction:Transfer-Ownership: transfer ownership 57 | FAC: facility 58 | GPE: geographical social political 59 | LOC: location 60 | ORG: organization 61 | PER: person 62 | VEH: vehicle 63 | WEA: weapon 64 | ORG-AFF: organization affiliation 65 | GEN-AFF: general affiliation 66 | PHYS: physical 67 | PART-WHOLE: part whole 68 | PER-SOC: personal social 69 | ART: agent artifact 70 | Defendant: defendant 71 | Prosecutor: prosecutor 72 | Person: person 73 | Origin: origin 74 | Buyer: buyer 75 | Plaintiff: plaintiff 76 | Victim: victim 77 | Org: organization 78 | Adjudicator: adjudicator 79 | Seller: seller 80 | Beneficiary: beneficiary 81 | Giver: giver 82 | Target: target 83 | Agent: agent 84 | Instrument: instrument 85 | Vehicle: vehicle 86 | Entity: entity 87 | Destination: destination 88 | Recipient: recipient 89 | Attacker: attacker 90 | Artifact: artifact 91 | Place: place 92 | 93 | 94 | 95 | 96 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R6.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - condemn 11 | - giveaway 12 | - remittance 13 | - pass away 14 | - go out of business 15 | - end 16 | - encounter 17 | - carry 18 | - parade 19 | - separate 20 | - attend the trial 21 | - perform 22 | - vote 23 | - begin 24 | - prosecute 25 | delete_list: 26 | mapper: 27 | Business:Declare-Bankruptcy: go out of business 28 | Business:End-Org: end organization 29 | Business:Merge-Org: merge organization 30 | Business:Start-Org: start organization 31 | Conflict:Attack: attack 32 | Conflict:Demonstrate: parade 33 | Contact:Meet: encounter 34 | Contact:Phone-Write: phone write 35 | Justice:Acquit: acquit 36 | Justice:Appeal: appeal 37 | Justice:Arrest-Jail: arrest jail 38 | Justice:Charge-Indict: prosecute 39 | Justice:Convict: convict 40 | Justice:Execute: perform 41 | Justice:Extradite: extradite 42 | Justice:Fine: fine 43 | Justice:Pardon: pardon 44 | Justice:Release-Parole: release parole 45 | Justice:Sentence: condemn 46 | Justice:Sue: sue 47 | Justice:Trial-Hearing: attend the trial 48 | Life:Be-Born: be born 49 | Life:Die: pass away 50 | Life:Divorce: separate 51 | Life:Injure: injure 52 | Life:Marry: marry 53 | Movement:Transport: carry 54 | Personnel:Elect: vote 55 | Personnel:End-Position: end 56 | Personnel:Nominate: nominate 57 | Personnel:Start-Position: begin 58 | Transaction:Transfer-Money: remittance 59 | Transaction:Transfer-Ownership: giveaway 60 | FAC: facility 61 | GPE: geographical social political 62 | LOC: location 63 | ORG: organization 64 | PER: person 65 | VEH: vehicle 66 | WEA: weapon 67 | ORG-AFF: organization affiliation 68 | GEN-AFF: general affiliation 69 | PHYS: physical 70 | PART-WHOLE: part whole 71 | PER-SOC: personal social 72 | ART: agent artifact 73 | Defendant: defendant 74 | Prosecutor: prosecutor 75 | Person: person 76 | Origin: origin 77 | Buyer: buyer 78 | Plaintiff: plaintiff 79 | Victim: victim 80 | Org: organization 81 | Adjudicator: adjudicator 82 | Seller: seller 83 | Beneficiary: beneficiary 84 | Giver: giver 85 | Target: target 86 | Agent: agent 87 | Instrument: instrument 88 | Vehicle: vehicle 89 | Entity: entity 90 | Destination: destination 91 | Recipient: recipient 92 | Attacker: attacker 93 | Artifact: artifact 94 | Place: place 95 | 96 | 97 | 98 | 99 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_R7.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | new_list: 10 | - giveaway 11 | - condemn 12 | - remittance 13 | - pass away 14 | - go out of business 15 | - wed 16 | - end 17 | - encounter 18 | - hurt 19 | - carry 20 | - parade 21 | - communication 22 | - separate 23 | - attend the trial 24 | - perform 25 | - vote 26 | - begin 27 | - prosecute 28 | delete_list: 29 | mapper: 30 | Business:Declare-Bankruptcy: go out of business 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: parade 36 | Contact:Meet: encounter 37 | Contact:Phone-Write: communication 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: prosecute 42 | Justice:Convict: convict 43 | Justice:Execute: perform 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: condemn 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: attend the trial 51 | Life:Be-Born: be born 52 | Life:Die: pass away 53 | Life:Divorce: separate 54 | Life:Injure: hurt 55 | Life:Marry: wed 56 | Movement:Transport: carry 57 | Personnel:Elect: vote 58 | Personnel:End-Position: end 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: begin 61 | Transaction:Transfer-Money: remittance 62 | Transaction:Transfer-Ownership: giveaway 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place 98 | 99 | 100 | 101 | 102 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V1.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | mapper: 12 | Business:Declare-Bankruptcy: business 13 | Business:End-Org: business 14 | Business:Merge-Org: business 15 | Business:Start-Org: business 16 | Conflict:Attack: attack 17 | Conflict:Demonstrate: demonstrate 18 | Contact:Meet: contact 19 | Contact:Phone-Write: contact 20 | Justice:Acquit: justice 21 | Justice:Appeal: justice 22 | Justice:Arrest-Jail: justice 23 | Justice:Charge-Indict: justice 24 | Justice:Convict: justice 25 | Justice:Execute: execute 26 | Justice:Extradite: extradite 27 | Justice:Fine: fine 28 | Justice:Pardon: pardon 29 | Justice:Release-Parole: release parole 30 | Justice:Sentence: sentence 31 | Justice:Sue: sue 32 | Justice:Trial-Hearing: justice 33 | Life:Be-Born: be born 34 | Life:Die: die 35 | Life:Divorce: life 36 | Life:Injure: life 37 | Life:Marry: marry 38 | Movement:Transport: transport 39 | Personnel:Elect: personnel 40 | Personnel:End-Position: personnel 41 | Personnel:Nominate: personnel 42 | Personnel:Start-Position: personnel 43 | Transaction:Transfer-Money: transfer money 44 | Transaction:Transfer-Ownership: transfer ownership 45 | FAC: facility 46 | GPE: geographical social political 47 | LOC: location 48 | ORG: organization 49 | PER: person 50 | VEH: vehicle 51 | WEA: weapon 52 | ORG-AFF: organization affiliation 53 | GEN-AFF: general affiliation 54 | PHYS: physical 55 | PART-WHOLE: part whole 56 | PER-SOC: personal social 57 | ART: agent artifact 58 | Defendant: defendant 59 | Prosecutor: prosecutor 60 | Person: person 61 | Origin: origin 62 | Buyer: buyer 63 | Plaintiff: plaintiff 64 | Victim: victim 65 | Org: organization 66 | Adjudicator: adjudicator 67 | Seller: seller 68 | Beneficiary: beneficiary 69 | Giver: giver 70 | Target: target 71 | Agent: agent 72 | Instrument: instrument 73 | Vehicle: vehicle 74 | Entity: entity 75 | Destination: destination 76 | Recipient: recipient 77 | Attacker: attacker 78 | Artifact: artifact 79 | Place: place 80 | 81 | 82 | 83 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V2.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - acquit 12 | - appeal 13 | - meet 14 | mapper: 15 | Business:Declare-Bankruptcy: business 16 | Business:End-Org: business 17 | Business:Merge-Org: business 18 | Business:Start-Org: business 19 | Conflict:Attack: attack 20 | Conflict:Demonstrate: demonstrate 21 | Contact:Meet: meet 22 | Contact:Phone-Write: contact 23 | Justice:Acquit: acquit 24 | Justice:Appeal: appeal 25 | Justice:Arrest-Jail: justice 26 | Justice:Charge-Indict: justice 27 | Justice:Convict: justice 28 | Justice:Execute: execute 29 | Justice:Extradite: extradite 30 | Justice:Fine: fine 31 | Justice:Pardon: pardon 32 | Justice:Release-Parole: release parole 33 | Justice:Sentence: sentence 34 | Justice:Sue: sue 35 | Justice:Trial-Hearing: justice 36 | Life:Be-Born: be born 37 | Life:Die: die 38 | Life:Divorce: life 39 | Life:Injure: life 40 | Life:Marry: marry 41 | Movement:Transport: transport 42 | Personnel:Elect: personnel 43 | Personnel:End-Position: personnel 44 | Personnel:Nominate: personnel 45 | Personnel:Start-Position: personnel 46 | Transaction:Transfer-Money: transfer money 47 | Transaction:Transfer-Ownership: transfer ownership 48 | FAC: facility 49 | GPE: geographical social political 50 | LOC: location 51 | ORG: organization 52 | PER: person 53 | VEH: vehicle 54 | WEA: weapon 55 | ORG-AFF: organization affiliation 56 | GEN-AFF: general affiliation 57 | PHYS: physical 58 | PART-WHOLE: part whole 59 | PER-SOC: personal social 60 | ART: agent artifact 61 | Defendant: defendant 62 | Prosecutor: prosecutor 63 | Person: person 64 | Origin: origin 65 | Buyer: buyer 66 | Plaintiff: plaintiff 67 | Victim: victim 68 | Org: organization 69 | Adjudicator: adjudicator 70 | Seller: seller 71 | Beneficiary: beneficiary 72 | Giver: giver 73 | Target: target 74 | Agent: agent 75 | Instrument: instrument 76 | Vehicle: vehicle 77 | Entity: entity 78 | Destination: destination 79 | Recipient: recipient 80 | Attacker: attacker 81 | Artifact: artifact 82 | Place: place 83 | 84 | 85 | 86 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V3.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - acquit 12 | - appeal 13 | - meet 14 | - merge organization 15 | - start organization 16 | - trial hearing 17 | mapper: 18 | Business:Declare-Bankruptcy: business 19 | Business:End-Org: business 20 | Business:Merge-Org: merge organization 21 | Business:Start-Org: start organization 22 | Conflict:Attack: attack 23 | Conflict:Demonstrate: demonstrate 24 | Contact:Meet: meet 25 | Contact:Phone-Write: contact 26 | Justice:Acquit: acquit 27 | Justice:Appeal: appeal 28 | Justice:Arrest-Jail: justice 29 | Justice:Charge-Indict: justice 30 | Justice:Convict: justice 31 | Justice:Execute: execute 32 | Justice:Extradite: extradite 33 | Justice:Fine: fine 34 | Justice:Pardon: pardon 35 | Justice:Release-Parole: release parole 36 | Justice:Sentence: sentence 37 | Justice:Sue: sue 38 | Justice:Trial-Hearing: trial hearing 39 | Life:Be-Born: be born 40 | Life:Die: die 41 | Life:Divorce: life 42 | Life:Injure: life 43 | Life:Marry: marry 44 | Movement:Transport: transport 45 | Personnel:Elect: personnel 46 | Personnel:End-Position: personnel 47 | Personnel:Nominate: personnel 48 | Personnel:Start-Position: personnel 49 | Transaction:Transfer-Money: transfer money 50 | Transaction:Transfer-Ownership: transfer ownership 51 | FAC: facility 52 | GPE: geographical social political 53 | LOC: location 54 | ORG: organization 55 | PER: person 56 | VEH: vehicle 57 | WEA: weapon 58 | ORG-AFF: organization affiliation 59 | GEN-AFF: general affiliation 60 | PHYS: physical 61 | PART-WHOLE: part whole 62 | PER-SOC: personal social 63 | ART: agent artifact 64 | Defendant: defendant 65 | Prosecutor: prosecutor 66 | Person: person 67 | Origin: origin 68 | Buyer: buyer 69 | Plaintiff: plaintiff 70 | Victim: victim 71 | Org: organization 72 | Adjudicator: adjudicator 73 | Seller: seller 74 | Beneficiary: beneficiary 75 | Giver: giver 76 | Target: target 77 | Agent: agent 78 | Instrument: instrument 79 | Vehicle: vehicle 80 | Entity: entity 81 | Destination: destination 82 | Recipient: recipient 83 | Attacker: attacker 84 | Artifact: artifact 85 | Place: place 86 | 87 | 88 | 89 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V4.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - acquit 12 | - appeal 13 | - declare bankruptcy 14 | - end organization 15 | - meet 16 | - merge organization 17 | - phone write 18 | - start organization 19 | - trial hearing 20 | mapper: 21 | Business:Declare-Bankruptcy: declare bankruptcy 22 | Business:End-Org: end organization 23 | Business:Merge-Org: merge organization 24 | Business:Start-Org: start organization 25 | Conflict:Attack: attack 26 | Conflict:Demonstrate: demonstrate 27 | Contact:Meet: meet 28 | Contact:Phone-Write: phone write 29 | Justice:Acquit: acquit 30 | Justice:Appeal: appeal 31 | Justice:Arrest-Jail: justice 32 | Justice:Charge-Indict: justice 33 | Justice:Convict: justice 34 | Justice:Execute: execute 35 | Justice:Extradite: extradite 36 | Justice:Fine: fine 37 | Justice:Pardon: pardon 38 | Justice:Release-Parole: release parole 39 | Justice:Sentence: sentence 40 | Justice:Sue: sue 41 | Justice:Trial-Hearing: trial hearing 42 | Life:Be-Born: be born 43 | Life:Die: die 44 | Life:Divorce: life 45 | Life:Injure: life 46 | Life:Marry: marry 47 | Movement:Transport: transport 48 | Personnel:Elect: personnel 49 | Personnel:End-Position: personnel 50 | Personnel:Nominate: personnel 51 | Personnel:Start-Position: personnel 52 | Transaction:Transfer-Money: transfer money 53 | Transaction:Transfer-Ownership: transfer ownership 54 | FAC: facility 55 | GPE: geographical social political 56 | LOC: location 57 | ORG: organization 58 | PER: person 59 | VEH: vehicle 60 | WEA: weapon 61 | ORG-AFF: organization affiliation 62 | GEN-AFF: general affiliation 63 | PHYS: physical 64 | PART-WHOLE: part whole 65 | PER-SOC: personal social 66 | ART: agent artifact 67 | Defendant: defendant 68 | Prosecutor: prosecutor 69 | Person: person 70 | Origin: origin 71 | Buyer: buyer 72 | Plaintiff: plaintiff 73 | Victim: victim 74 | Org: organization 75 | Adjudicator: adjudicator 76 | Seller: seller 77 | Beneficiary: beneficiary 78 | Giver: giver 79 | Target: target 80 | Agent: agent 81 | Instrument: instrument 82 | Vehicle: vehicle 83 | Entity: entity 84 | Destination: destination 85 | Recipient: recipient 86 | Attacker: attacker 87 | Artifact: artifact 88 | Place: place 89 | 90 | 91 | 92 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V5.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - acquit 12 | - appeal 13 | - arrest jail 14 | - charge indict 15 | - convict 16 | - declare bankruptcy 17 | - end organization 18 | - meet 19 | - merge organization 20 | - phone write 21 | - start organization 22 | - trial hearing 23 | mapper: 24 | Business:Declare-Bankruptcy: declare bankruptcy 25 | Business:End-Org: end organization 26 | Business:Merge-Org: merge organization 27 | Business:Start-Org: start organization 28 | Conflict:Attack: attack 29 | Conflict:Demonstrate: demonstrate 30 | Contact:Meet: meet 31 | Contact:Phone-Write: phone write 32 | Justice:Acquit: acquit 33 | Justice:Appeal: appeal 34 | Justice:Arrest-Jail: arrest jail 35 | Justice:Charge-Indict: charge indict 36 | Justice:Convict: convict 37 | Justice:Execute: execute 38 | Justice:Extradite: extradite 39 | Justice:Fine: fine 40 | Justice:Pardon: pardon 41 | Justice:Release-Parole: release parole 42 | Justice:Sentence: sentence 43 | Justice:Sue: sue 44 | Justice:Trial-Hearing: trial hearing 45 | Life:Be-Born: be born 46 | Life:Die: die 47 | Life:Divorce: life 48 | Life:Injure: life 49 | Life:Marry: marry 50 | Movement:Transport: transport 51 | Personnel:Elect: personnel 52 | Personnel:End-Position: personnel 53 | Personnel:Nominate: personnel 54 | Personnel:Start-Position: personnel 55 | Transaction:Transfer-Money: transfer money 56 | Transaction:Transfer-Ownership: transfer ownership 57 | FAC: facility 58 | GPE: geographical social political 59 | LOC: location 60 | ORG: organization 61 | PER: person 62 | VEH: vehicle 63 | WEA: weapon 64 | ORG-AFF: organization affiliation 65 | GEN-AFF: general affiliation 66 | PHYS: physical 67 | PART-WHOLE: part whole 68 | PER-SOC: personal social 69 | ART: agent artifact 70 | Defendant: defendant 71 | Prosecutor: prosecutor 72 | Person: person 73 | Origin: origin 74 | Buyer: buyer 75 | Plaintiff: plaintiff 76 | Victim: victim 77 | Org: organization 78 | Adjudicator: adjudicator 79 | Seller: seller 80 | Beneficiary: beneficiary 81 | Giver: giver 82 | Target: target 83 | Agent: agent 84 | Instrument: instrument 85 | Vehicle: vehicle 86 | Entity: entity 87 | Destination: destination 88 | Recipient: recipient 89 | Attacker: attacker 90 | Artifact: artifact 91 | Place: place 92 | 93 | 94 | 95 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V6.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - acquit 12 | - appeal 13 | - arrest jail 14 | - charge indict 15 | - convict 16 | - declare bankruptcy 17 | - divorce 18 | - end organization 19 | - end position 20 | - injure 21 | - meet 22 | - merge organization 23 | - phone write 24 | - start organization 25 | - trial hearing 26 | mapper: 27 | Business:Declare-Bankruptcy: declare bankruptcy 28 | Business:End-Org: end organization 29 | Business:Merge-Org: merge organization 30 | Business:Start-Org: start organization 31 | Conflict:Attack: attack 32 | Conflict:Demonstrate: demonstrate 33 | Contact:Meet: meet 34 | Contact:Phone-Write: phone write 35 | Justice:Acquit: acquit 36 | Justice:Appeal: appeal 37 | Justice:Arrest-Jail: arrest jail 38 | Justice:Charge-Indict: charge indict 39 | Justice:Convict: convict 40 | Justice:Execute: execute 41 | Justice:Extradite: extradite 42 | Justice:Fine: fine 43 | Justice:Pardon: pardon 44 | Justice:Release-Parole: release parole 45 | Justice:Sentence: sentence 46 | Justice:Sue: sue 47 | Justice:Trial-Hearing: trial hearing 48 | Life:Be-Born: be born 49 | Life:Die: die 50 | Life:Divorce: divorce 51 | Life:Injure: injure 52 | Life:Marry: marry 53 | Movement:Transport: transport 54 | Personnel:Elect: personnel 55 | Personnel:End-Position: end position 56 | Personnel:Nominate: personnel 57 | Personnel:Start-Position: personnel 58 | Transaction:Transfer-Money: transfer money 59 | Transaction:Transfer-Ownership: transfer ownership 60 | FAC: facility 61 | GPE: geographical social political 62 | LOC: location 63 | ORG: organization 64 | PER: person 65 | VEH: vehicle 66 | WEA: weapon 67 | ORG-AFF: organization affiliation 68 | GEN-AFF: general affiliation 69 | PHYS: physical 70 | PART-WHOLE: part whole 71 | PER-SOC: personal social 72 | ART: agent artifact 73 | Defendant: defendant 74 | Prosecutor: prosecutor 75 | Person: person 76 | Origin: origin 77 | Buyer: buyer 78 | Plaintiff: plaintiff 79 | Victim: victim 80 | Org: organization 81 | Adjudicator: adjudicator 82 | Seller: seller 83 | Beneficiary: beneficiary 84 | Giver: giver 85 | Target: target 86 | Agent: agent 87 | Instrument: instrument 88 | Vehicle: vehicle 89 | Entity: entity 90 | Destination: destination 91 | Recipient: recipient 92 | Attacker: attacker 93 | Artifact: artifact 94 | Place: place 95 | 96 | 97 | 98 | -------------------------------------------------------------------------------- /config/data_config/event/ace05_event_V7.yaml: -------------------------------------------------------------------------------- 1 | name: ace05_event 2 | path: data/oneie/ace05-EN 3 | data_class: OneIEEvent 4 | split: 5 | train: train.oneie.json 6 | val: dev.oneie.json 7 | test: test.oneie.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - acquit 12 | - appeal 13 | - arrest jail 14 | - charge indict 15 | - convict 16 | - declare bankruptcy 17 | - divorce 18 | - elect 19 | - end organization 20 | - end position 21 | - injure 22 | - meet 23 | - merge organization 24 | - nominate 25 | - phone write 26 | - start organization 27 | - start position 28 | - trial hearing 29 | mapper: 30 | Business:Declare-Bankruptcy: declare bankruptcy 31 | Business:End-Org: end organization 32 | Business:Merge-Org: merge organization 33 | Business:Start-Org: start organization 34 | Conflict:Attack: attack 35 | Conflict:Demonstrate: demonstrate 36 | Contact:Meet: meet 37 | Contact:Phone-Write: phone write 38 | Justice:Acquit: acquit 39 | Justice:Appeal: appeal 40 | Justice:Arrest-Jail: arrest jail 41 | Justice:Charge-Indict: charge indict 42 | Justice:Convict: convict 43 | Justice:Execute: execute 44 | Justice:Extradite: extradite 45 | Justice:Fine: fine 46 | Justice:Pardon: pardon 47 | Justice:Release-Parole: release parole 48 | Justice:Sentence: sentence 49 | Justice:Sue: sue 50 | Justice:Trial-Hearing: trial hearing 51 | Life:Be-Born: be born 52 | Life:Die: die 53 | Life:Divorce: divorce 54 | Life:Injure: injure 55 | Life:Marry: marry 56 | Movement:Transport: transport 57 | Personnel:Elect: elect 58 | Personnel:End-Position: end position 59 | Personnel:Nominate: nominate 60 | Personnel:Start-Position: start position 61 | Transaction:Transfer-Money: transfer money 62 | Transaction:Transfer-Ownership: transfer ownership 63 | FAC: facility 64 | GPE: geographical social political 65 | LOC: location 66 | ORG: organization 67 | PER: person 68 | VEH: vehicle 69 | WEA: weapon 70 | ORG-AFF: organization affiliation 71 | GEN-AFF: general affiliation 72 | PHYS: physical 73 | PART-WHOLE: part whole 74 | PER-SOC: personal social 75 | ART: agent artifact 76 | Defendant: defendant 77 | Prosecutor: prosecutor 78 | Person: person 79 | Origin: origin 80 | Buyer: buyer 81 | Plaintiff: plaintiff 82 | Victim: victim 83 | Org: organization 84 | Adjudicator: adjudicator 85 | Seller: seller 86 | Beneficiary: beneficiary 87 | Giver: giver 88 | Target: target 89 | Agent: agent 90 | Instrument: instrument 91 | Vehicle: vehicle 92 | Entity: entity 93 | Destination: destination 94 | Recipient: recipient 95 | Attacker: attacker 96 | Artifact: artifact 97 | Place: place 98 | 99 | 100 | 101 | -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H1.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | delete_list: 10 | - /people/person/ethnicity 11 | - /location/country/administrative_divisions 12 | - /business/person/company 13 | - /people/person/religion 14 | - /location/administrative_division/country 15 | - /people/ethnicity/geographic_distribution 16 | - /business/company/major_shareholders 17 | - /people/person/place_lived 18 | - /business/company/founders 19 | - /location/country/capital 20 | - /location/neighborhood/neighborhood_of 21 | - /sports/sports_team/location 22 | - /people/person/nationality 23 | - /people/person/place_of_birth 24 | mapper: 25 | /business/company/advisors: advisors 26 | /business/company/founders: founders 27 | /business/company/industry: industry 28 | /business/company/major_shareholders: major shareholders 29 | /business/company/place_founded: place founded 30 | /business/company_shareholder/major_shareholder_of: major shareholder of 31 | /business/person/company: company 32 | /location/administrative_division/country: country 33 | /location/country/administrative_divisions: administrative divisions 34 | /location/country/capital: capital 35 | /location/location/contains: contains 36 | /location/neighborhood/neighborhood_of: neighborhood of 37 | /people/deceased_person/place_of_death: place of death 38 | /people/ethnicity/geographic_distribution: geographic distribution 39 | /people/ethnicity/people: people 40 | /people/person/children: children 41 | /people/person/ethnicity: ethnicity 42 | /people/person/nationality: nationality 43 | /people/person/place_lived: place lived 44 | /people/person/place_of_birth: brith place 45 | /people/person/profession: profession 46 | /people/person/religion: religion 47 | /sports/sports_team/location: location 48 | /sports/sports_team_location/teams: teams 49 | ORGANIZATION: organization 50 | LOCATION: location 51 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H2.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - nationality 12 | delete_list: 13 | - /people/person/ethnicity 14 | - /location/country/administrative_divisions 15 | - /business/person/company 16 | - /people/person/religion 17 | - /location/administrative_division/country 18 | - /people/ethnicity/geographic_distribution 19 | - /business/company/major_shareholders 20 | - /people/person/place_lived 21 | - /business/company/founders 22 | - /location/country/capital 23 | - /location/neighborhood/neighborhood_of 24 | - /sports/sports_team/location 25 | mapper: 26 | /business/company/advisors: advisors 27 | /business/company/founders: founders 28 | /business/company/industry: industry 29 | /business/company/major_shareholders: major shareholders 30 | /business/company/place_founded: place founded 31 | /business/company_shareholder/major_shareholder_of: major shareholder of 32 | /business/person/company: company 33 | /location/administrative_division/country: country 34 | /location/country/administrative_divisions: administrative divisions 35 | /location/country/capital: capital 36 | /location/location/contains: contains 37 | /location/neighborhood/neighborhood_of: neighborhood of 38 | /people/deceased_person/place_of_death: place of death 39 | /people/ethnicity/geographic_distribution: geographic distribution 40 | /people/ethnicity/people: people 41 | /people/person/children: children 42 | /people/person/ethnicity: ethnicity 43 | /people/person/nationality: nationality 44 | /people/person/place_lived: place lived 45 | /people/person/place_of_birth: brith place 46 | /people/person/profession: profession 47 | /people/person/religion: religion 48 | /sports/sports_team/location: location 49 | /sports/sports_team_location/teams: teams 50 | ORGANIZATION: organization 51 | LOCATION: location 52 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H3.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - neighborhood of 11 | - brith place 12 | - nationality 13 | - location 14 | delete_list: 15 | - /people/person/ethnicity 16 | - /location/country/administrative_divisions 17 | - /business/person/company 18 | - /people/person/religion 19 | - /location/administrative_division/country 20 | - /people/ethnicity/geographic_distribution 21 | - /business/company/major_shareholders 22 | - /people/person/place_lived 23 | - /business/company/founders 24 | - /location/country/capital 25 | mapper: 26 | /business/company/advisors: advisors 27 | /business/company/founders: founders 28 | /business/company/industry: industry 29 | /business/company/major_shareholders: major shareholders 30 | /business/company/place_founded: place founded 31 | /business/company_shareholder/major_shareholder_of: major shareholder of 32 | /business/person/company: company 33 | /location/administrative_division/country: country 34 | /location/country/administrative_divisions: administrative divisions 35 | /location/country/capital: capital 36 | /location/location/contains: contains 37 | /location/neighborhood/neighborhood_of: neighborhood of 38 | /people/deceased_person/place_of_death: place of death 39 | /people/ethnicity/geographic_distribution: geographic distribution 40 | /people/ethnicity/people: people 41 | /people/person/children: children 42 | /people/person/ethnicity: ethnicity 43 | /people/person/nationality: nationality 44 | /people/person/place_lived: place lived 45 | /people/person/place_of_birth: brith place 46 | /people/person/profession: profession 47 | /people/person/religion: religion 48 | /sports/sports_team/location: location 49 | /sports/sports_team_location/teams: teams 50 | ORGANIZATION: organization 51 | LOCATION: location 52 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H4.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - neighborhood of 12 | - capital 13 | - nationality 14 | - founders 15 | - location 16 | delete_list: 17 | - /people/person/ethnicity 18 | - /location/country/administrative_divisions 19 | - /business/person/company 20 | - /people/person/religion 21 | - /location/administrative_division/country 22 | - /people/ethnicity/geographic_distribution 23 | - /business/company/major_shareholders 24 | - /people/person/place_lived 25 | mapper: 26 | /business/company/advisors: advisors 27 | /business/company/founders: founders 28 | /business/company/industry: industry 29 | /business/company/major_shareholders: major shareholders 30 | /business/company/place_founded: place founded 31 | /business/company_shareholder/major_shareholder_of: major shareholder of 32 | /business/person/company: company 33 | /location/administrative_division/country: country 34 | /location/country/administrative_divisions: administrative divisions 35 | /location/country/capital: capital 36 | /location/location/contains: contains 37 | /location/neighborhood/neighborhood_of: neighborhood of 38 | /people/deceased_person/place_of_death: place of death 39 | /people/ethnicity/geographic_distribution: geographic distribution 40 | /people/ethnicity/people: people 41 | /people/person/children: children 42 | /people/person/ethnicity: ethnicity 43 | /people/person/nationality: nationality 44 | /people/person/place_lived: place lived 45 | /people/person/place_of_birth: brith place 46 | /people/person/profession: profession 47 | /people/person/religion: religion 48 | /sports/sports_team/location: location 49 | /sports/sports_team_location/teams: teams 50 | ORGANIZATION: organization 51 | LOCATION: location 52 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H5.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - place lived 12 | - neighborhood of 13 | - capital 14 | - major shareholders 15 | - nationality 16 | - founders 17 | - location 18 | delete_list: 19 | - /people/person/ethnicity 20 | - /location/country/administrative_divisions 21 | - /business/person/company 22 | - /people/person/religion 23 | - /location/administrative_division/country 24 | - /people/ethnicity/geographic_distribution 25 | mapper: 26 | /business/company/advisors: advisors 27 | /business/company/founders: founders 28 | /business/company/industry: industry 29 | /business/company/major_shareholders: major shareholders 30 | /business/company/place_founded: place founded 31 | /business/company_shareholder/major_shareholder_of: major shareholder of 32 | /business/person/company: company 33 | /location/administrative_division/country: country 34 | /location/country/administrative_divisions: administrative divisions 35 | /location/country/capital: capital 36 | /location/location/contains: contains 37 | /location/neighborhood/neighborhood_of: neighborhood of 38 | /people/deceased_person/place_of_death: place of death 39 | /people/ethnicity/geographic_distribution: geographic distribution 40 | /people/ethnicity/people: people 41 | /people/person/children: children 42 | /people/person/ethnicity: ethnicity 43 | /people/person/nationality: nationality 44 | /people/person/place_lived: place lived 45 | /people/person/place_of_birth: brith place 46 | /people/person/profession: profession 47 | /people/person/religion: religion 48 | /sports/sports_team/location: location 49 | /sports/sports_team_location/teams: teams 50 | ORGANIZATION: organization 51 | LOCATION: location 52 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H6.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - place lived 12 | - neighborhood of 13 | - capital 14 | - major shareholders 15 | - country 16 | - geographic distribution 17 | - nationality 18 | - founders 19 | - location 20 | delete_list: 21 | - /people/person/ethnicity 22 | - /location/country/administrative_divisions 23 | - /business/person/company 24 | - /people/person/religion 25 | mapper: 26 | /business/company/advisors: advisors 27 | /business/company/founders: founders 28 | /business/company/industry: industry 29 | /business/company/major_shareholders: major shareholders 30 | /business/company/place_founded: place founded 31 | /business/company_shareholder/major_shareholder_of: major shareholder of 32 | /business/person/company: company 33 | /location/administrative_division/country: country 34 | /location/country/administrative_divisions: administrative divisions 35 | /location/country/capital: capital 36 | /location/location/contains: contains 37 | /location/neighborhood/neighborhood_of: neighborhood of 38 | /people/deceased_person/place_of_death: place of death 39 | /people/ethnicity/geographic_distribution: geographic distribution 40 | /people/ethnicity/people: people 41 | /people/person/children: children 42 | /people/person/ethnicity: ethnicity 43 | /people/person/nationality: nationality 44 | /people/person/place_lived: place lived 45 | /people/person/place_of_birth: brith place 46 | /people/person/profession: profession 47 | /people/person/religion: religion 48 | /sports/sports_team/location: location 49 | /sports/sports_team_location/teams: teams 50 | ORGANIZATION: organization 51 | LOCATION: location 52 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_H7.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - capital 11 | - neighborhood of 12 | - geographic distribution 13 | - brith place 14 | - place lived 15 | - country 16 | - major shareholders 17 | - nationality 18 | - religion 19 | - company 20 | - founders 21 | - location 22 | delete_list: 23 | - /people/person/ethnicity 24 | - /location/country/administrative_divisions 25 | mapper: 26 | /business/company/advisors: advisors 27 | /business/company/founders: founders 28 | /business/company/industry: industry 29 | /business/company/major_shareholders: major shareholders 30 | /business/company/place_founded: place founded 31 | /business/company_shareholder/major_shareholder_of: major shareholder of 32 | /business/person/company: company 33 | /location/administrative_division/country: country 34 | /location/country/administrative_divisions: administrative divisions 35 | /location/country/capital: capital 36 | /location/location/contains: contains 37 | /location/neighborhood/neighborhood_of: neighborhood of 38 | /people/deceased_person/place_of_death: place of death 39 | /people/ethnicity/geographic_distribution: geographic distribution 40 | /people/ethnicity/people: people 41 | /people/person/children: children 42 | /people/person/ethnicity: ethnicity 43 | /people/person/nationality: nationality 44 | /people/person/place_lived: place lived 45 | /people/person/place_of_birth: brith place 46 | /people/person/profession: profession 47 | /people/person/religion: religion 48 | /sports/sports_team/location: location 49 | /sports/sports_team_location/teams: teams 50 | ORGANIZATION: organization 51 | LOCATION: location 52 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M1.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | delete_list: 11 | - /sports/sports_team_location/teams 12 | mapper: 13 | /business/company/advisors: advisors 14 | /business/company/founders: business 15 | /business/company/industry: industry 16 | /business/company/major_shareholders: business 17 | /business/company/place_founded: place founded 18 | /business/company_shareholder/major_shareholder_of: major shareholder of 19 | /business/person/company: company 20 | /location/administrative_division/country: location 21 | /location/country/administrative_divisions: location 22 | /location/country/capital: location 23 | /location/location/contains: contains 24 | /location/neighborhood/neighborhood_of: location 25 | /people/deceased_person/place_of_death: people 26 | /people/ethnicity/geographic_distribution: geographic distribution 27 | /people/ethnicity/people: people group 28 | /people/person/children: children 29 | /people/person/ethnicity: ethnicity 30 | /people/person/nationality: people 31 | /people/person/place_lived: people 32 | /people/person/place_of_birth: people 33 | /people/person/profession: profession 34 | /people/person/religion: religion 35 | /sports/sports_team/location: sports 36 | /sports/sports_team_location/teams: teams 37 | ORGANIZATION: organization 38 | LOCATION: location 39 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M2.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - country 11 | - founders 12 | delete_list: 13 | - /sports/sports_team_location/teams 14 | mapper: 15 | /business/company/advisors: advisors 16 | /business/company/founders: founders 17 | /business/company/industry: industry 18 | /business/company/major_shareholders: business 19 | /business/company/place_founded: place founded 20 | /business/company_shareholder/major_shareholder_of: major shareholder of 21 | /business/person/company: company 22 | /location/administrative_division/country: country 23 | /location/country/administrative_divisions: location 24 | /location/country/capital: location 25 | /location/location/contains: contains 26 | /location/neighborhood/neighborhood_of: location 27 | /people/deceased_person/place_of_death: people 28 | /people/ethnicity/geographic_distribution: geographic distribution 29 | /people/ethnicity/people: people group 30 | /people/person/children: children 31 | /people/person/ethnicity: ethnicity 32 | /people/person/nationality: people 33 | /people/person/place_lived: people 34 | /people/person/place_of_birth: people 35 | /people/person/profession: profession 36 | /people/person/religion: religion 37 | /sports/sports_team/location: sports 38 | /sports/sports_team_location/teams: teams 39 | ORGANIZATION: organization 40 | LOCATION: location 41 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M3.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - country 11 | - founders 12 | - neighborhood of 13 | - place of death 14 | delete_list: 15 | - /sports/sports_team_location/teams 16 | mapper: 17 | /business/company/advisors: advisors 18 | /business/company/founders: founders 19 | /business/company/industry: industry 20 | /business/company/major_shareholders: business 21 | /business/company/place_founded: place founded 22 | /business/company_shareholder/major_shareholder_of: major shareholder of 23 | /business/person/company: company 24 | /location/administrative_division/country: country 25 | /location/country/administrative_divisions: location 26 | /location/country/capital: location 27 | /location/location/contains: contains 28 | /location/neighborhood/neighborhood_of: neighborhood of 29 | /people/deceased_person/place_of_death: place of death 30 | /people/ethnicity/geographic_distribution: geographic distribution 31 | /people/ethnicity/people: people group 32 | /people/person/children: children 33 | /people/person/ethnicity: ethnicity 34 | /people/person/nationality: people 35 | /people/person/place_lived: people 36 | /people/person/place_of_birth: people 37 | /people/person/profession: profession 38 | /people/person/religion: religion 39 | /sports/sports_team/location: sports 40 | /sports/sports_team_location/teams: teams 41 | ORGANIZATION: organization 42 | LOCATION: location 43 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M4.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - country 12 | - founders 13 | - neighborhood of 14 | - place of death 15 | - teams 16 | delete_list: 17 | mapper: 18 | /business/company/advisors: advisors 19 | /business/company/founders: founders 20 | /business/company/industry: industry 21 | /business/company/major_shareholders: business 22 | /business/company/place_founded: place founded 23 | /business/company_shareholder/major_shareholder_of: major shareholder of 24 | /business/person/company: company 25 | /location/administrative_division/country: country 26 | /location/country/administrative_divisions: administrative divisions 27 | /location/country/capital: location 28 | /location/location/contains: contains 29 | /location/neighborhood/neighborhood_of: neighborhood of 30 | /people/deceased_person/place_of_death: place of death 31 | /people/ethnicity/geographic_distribution: geographic distribution 32 | /people/ethnicity/people: people group 33 | /people/person/children: children 34 | /people/person/ethnicity: ethnicity 35 | /people/person/nationality: people 36 | /people/person/place_lived: people 37 | /people/person/place_of_birth: people 38 | /people/person/profession: profession 39 | /people/person/religion: religion 40 | /sports/sports_team/location: sports 41 | /sports/sports_team_location/teams: teams 42 | ORGANIZATION: organization 43 | LOCATION: location 44 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M5.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - capital 12 | - country 13 | - founders 14 | - major shareholders 15 | - neighborhood of 16 | - place of death 17 | - teams 18 | delete_list: 19 | mapper: 20 | /business/company/advisors: advisors 21 | /business/company/founders: founders 22 | /business/company/industry: industry 23 | /business/company/major_shareholders: major shareholders 24 | /business/company/place_founded: place founded 25 | /business/company_shareholder/major_shareholder_of: major shareholder of 26 | /business/person/company: company 27 | /location/administrative_division/country: country 28 | /location/country/administrative_divisions: administrative divisions 29 | /location/country/capital: capital 30 | /location/location/contains: contains 31 | /location/neighborhood/neighborhood_of: neighborhood of 32 | /people/deceased_person/place_of_death: place of death 33 | /people/ethnicity/geographic_distribution: geographic distribution 34 | /people/ethnicity/people: people group 35 | /people/person/children: children 36 | /people/person/ethnicity: ethnicity 37 | /people/person/nationality: people 38 | /people/person/place_lived: people 39 | /people/person/place_of_birth: people 40 | /people/person/profession: profession 41 | /people/person/religion: religion 42 | /sports/sports_team/location: sports 43 | /sports/sports_team_location/teams: teams 44 | ORGANIZATION: organization 45 | LOCATION: location 46 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M6.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - capital 12 | - country 13 | - founders 14 | - location place 15 | - major shareholders 16 | - nationality 17 | - neighborhood of 18 | - place of death 19 | - teams 20 | delete_list: 21 | mapper: 22 | /business/company/advisors: advisors 23 | /business/company/founders: founders 24 | /business/company/industry: industry 25 | /business/company/major_shareholders: major shareholders 26 | /business/company/place_founded: place founded 27 | /business/company_shareholder/major_shareholder_of: major shareholder of 28 | /business/person/company: company 29 | /location/administrative_division/country: country 30 | /location/country/administrative_divisions: administrative divisions 31 | /location/country/capital: capital 32 | /location/location/contains: contains 33 | /location/neighborhood/neighborhood_of: neighborhood of 34 | /people/deceased_person/place_of_death: place of death 35 | /people/ethnicity/geographic_distribution: geographic distribution 36 | /people/ethnicity/people: people group 37 | /people/person/children: children 38 | /people/person/ethnicity: ethnicity 39 | /people/person/nationality: nationality 40 | /people/person/place_lived: people 41 | /people/person/place_of_birth: people 42 | /people/person/profession: profession 43 | /people/person/religion: religion 44 | /sports/sports_team/location: location place 45 | /sports/sports_team_location/teams: teams 46 | ORGANIZATION: organization 47 | LOCATION: location 48 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_M7.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - birth place 12 | - capital 13 | - country 14 | - founders 15 | - location place 16 | - major shareholders 17 | - nationality 18 | - neighborhood of 19 | - place lived 20 | - place of death 21 | - teams 22 | delete_list: 23 | mapper: 24 | /business/company/advisors: advisors 25 | /business/company/founders: founders 26 | /business/company/industry: industry 27 | /business/company/major_shareholders: major shareholders 28 | /business/company/place_founded: place founded 29 | /business/company_shareholder/major_shareholder_of: major shareholder of 30 | /business/person/company: company 31 | /location/administrative_division/country: country 32 | /location/country/administrative_divisions: administrative divisions 33 | /location/country/capital: capital 34 | /location/location/contains: contains 35 | /location/neighborhood/neighborhood_of: neighborhood of 36 | /people/deceased_person/place_of_death: place of death 37 | /people/ethnicity/geographic_distribution: geographic distribution 38 | /people/ethnicity/people: people group 39 | /people/person/children: children 40 | /people/person/ethnicity: ethnicity 41 | /people/person/nationality: nationality 42 | /people/person/place_lived: place lived 43 | /people/person/place_of_birth: birth place 44 | /people/person/profession: profession 45 | /people/person/religion: religion 46 | /sports/sports_team/location: location place 47 | /sports/sports_team_location/teams: teams 48 | ORGANIZATION: organization 49 | LOCATION: location 50 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R1.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | delete_list: 11 | mapper: 12 | /business/company/advisors: advisors 13 | /business/company/founders: founders 14 | /business/company/industry: industry 15 | /business/company/major_shareholders: major shareholders 16 | /business/company/place_founded: place founded 17 | /business/company_shareholder/major_shareholder_of: major shareholder of 18 | /business/person/company: company 19 | /location/administrative_division/country: country 20 | /location/country/administrative_divisions: administrative divisions 21 | /location/country/capital: capital 22 | /location/location/contains: contains 23 | /location/neighborhood/neighborhood_of: neighborhood of 24 | /people/deceased_person/place_of_death: place of death 25 | /people/ethnicity/geographic_distribution: geographic distribution 26 | /people/ethnicity/people: people 27 | /people/person/children: children 28 | /people/person/ethnicity: ethnicity 29 | /people/person/nationality: nationality 30 | /people/person/place_lived: place lived 31 | /people/person/place_of_birth: place of birth 32 | /people/person/profession: profession 33 | /people/person/religion: religion 34 | /sports/sports_team/location: location 35 | /sports/sports_team_location/teams: teams 36 | ORGANIZATION: organization 37 | LOCATION: location 38 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R2.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - corporation 12 | delete_list: 13 | mapper: 14 | /business/company/advisors: advisors 15 | /business/company/founders: founders 16 | /business/company/industry: industry 17 | /business/company/major_shareholders: major shareholders 18 | /business/company/place_founded: place founded 19 | /business/company_shareholder/major_shareholder_of: major shareholder of 20 | /business/person/company: corporation 21 | /location/administrative_division/country: country 22 | /location/country/administrative_divisions: administrative divisions 23 | /location/country/capital: capital 24 | /location/location/contains: contains 25 | /location/neighborhood/neighborhood_of: neighborhood of 26 | /people/deceased_person/place_of_death: place of death 27 | /people/ethnicity/geographic_distribution: geographic distribution 28 | /people/ethnicity/people: people 29 | /people/person/children: children 30 | /people/person/ethnicity: ethnicity 31 | /people/person/nationality: nationality 32 | /people/person/place_lived: place lived 33 | /people/person/place_of_birth: brith place 34 | /people/person/profession: profession 35 | /people/person/religion: religion 36 | /sports/sports_team/location: location 37 | /sports/sports_team_location/teams: teams 38 | ORGANIZATION: organization 39 | LOCATION: location 40 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R3.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - membership of a nation 11 | - persons in general 12 | - brith place 13 | - corporation 14 | delete_list: 15 | mapper: 16 | /business/company/advisors: advisors 17 | /business/company/founders: founders 18 | /business/company/industry: industry 19 | /business/company/major_shareholders: major shareholders 20 | /business/company/place_founded: place founded 21 | /business/company_shareholder/major_shareholder_of: major shareholder of 22 | /business/person/company: corporation 23 | /location/administrative_division/country: country 24 | /location/country/administrative_divisions: administrative divisions 25 | /location/country/capital: capital 26 | /location/location/contains: contains 27 | /location/neighborhood/neighborhood_of: neighborhood of 28 | /people/deceased_person/place_of_death: place of death 29 | /people/ethnicity/geographic_distribution: geographic distribution 30 | /people/ethnicity/people: persons in general 31 | /people/person/children: children 32 | /people/person/ethnicity: ethnicity 33 | /people/person/nationality: membership of a nation 34 | /people/person/place_lived: place lived 35 | /people/person/place_of_birth: brith place 36 | /people/person/profession: profession 37 | /people/person/religion: religion 38 | /sports/sports_team/location: location 39 | /sports/sports_team_location/teams: teams 40 | ORGANIZATION: organization 41 | LOCATION: location 42 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R4.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - membership of a nation 12 | - seat of government 13 | - originator 14 | - persons in general 15 | - corporation 16 | delete_list: 17 | mapper: 18 | /business/company/advisors: advisors 19 | /business/company/founders: originator 20 | /business/company/industry: industry 21 | /business/company/major_shareholders: major shareholders 22 | /business/company/place_founded: place founded 23 | /business/company_shareholder/major_shareholder_of: major shareholder of 24 | /business/person/company: corporation 25 | /location/administrative_division/country: country 26 | /location/country/administrative_divisions: administrative divisions 27 | /location/country/capital: seat of government 28 | /location/location/contains: contains 29 | /location/neighborhood/neighborhood_of: neighborhood of 30 | /people/deceased_person/place_of_death: place of death 31 | /people/ethnicity/geographic_distribution: geographic distribution 32 | /people/ethnicity/people: persons in general 33 | /people/person/children: children 34 | /people/person/ethnicity: ethnicity 35 | /people/person/nationality: membership of a nation 36 | /people/person/place_lived: place lived 37 | /people/person/place_of_birth: brith place 38 | /people/person/profession: profession 39 | /people/person/religion: religion 40 | /sports/sports_team/location: location 41 | /sports/sports_team_location/teams: teams 42 | ORGANIZATION: organization 43 | LOCATION: location 44 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R5.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - brith place 11 | - state 12 | - membership of a nation 13 | - consultant 14 | - seat of government 15 | - originator 16 | - persons in general 17 | - corporation 18 | delete_list: 19 | mapper: 20 | /business/company/advisors: consultant 21 | /business/company/founders: originator 22 | /business/company/industry: industry 23 | /business/company/major_shareholders: major shareholders 24 | /business/company/place_founded: place founded 25 | /business/company_shareholder/major_shareholder_of: major shareholder of 26 | /business/person/company: corporation 27 | /location/administrative_division/country: state 28 | /location/country/administrative_divisions: administrative divisions 29 | /location/country/capital: seat of government 30 | /location/location/contains: contains 31 | /location/neighborhood/neighborhood_of: neighborhood of 32 | /people/deceased_person/place_of_death: place of death 33 | /people/ethnicity/geographic_distribution: geographic distribution 34 | /people/ethnicity/people: persons in general 35 | /people/person/children: children 36 | /people/person/ethnicity: ethnicity 37 | /people/person/nationality: membership of a nation 38 | /people/person/place_lived: place lived 39 | /people/person/place_of_birth: brith place 40 | /people/person/profession: profession 41 | /people/person/religion: religion 42 | /sports/sports_team/location: location 43 | /sports/sports_team_location/teams: teams 44 | ORGANIZATION: organization 45 | LOCATION: location 46 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R6.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - background 11 | - brith place 12 | - state 13 | - membership of a nation 14 | - consultant 15 | - business firm 16 | - seat of government 17 | - originator 18 | - persons in general 19 | - corporation 20 | delete_list: 21 | mapper: 22 | /business/company/advisors: consultant 23 | /business/company/founders: originator 24 | /business/company/industry: industry 25 | /business/company/major_shareholders: major shareholders 26 | /business/company/place_founded: place founded 27 | /business/company_shareholder/major_shareholder_of: major shareholder of 28 | /business/person/company: business firm 29 | /location/administrative_division/country: state 30 | /location/country/administrative_divisions: administrative divisions 31 | /location/country/capital: seat of government 32 | /location/location/contains: contains 33 | /location/neighborhood/neighborhood_of: neighborhood of 34 | /people/deceased_person/place_of_death: place of death 35 | /people/ethnicity/geographic_distribution: geographic distribution 36 | /people/ethnicity/people: persons in general 37 | /people/person/children: children 38 | /people/person/ethnicity: background 39 | /people/person/nationality: membership of a nation 40 | /people/person/place_lived: place lived 41 | /people/person/place_of_birth: brith place 42 | /people/person/profession: profession 43 | /people/person/religion: religion 44 | /sports/sports_team/location: location 45 | /sports/sports_team_location/teams: teams 46 | ORGANIZATION: organization 47 | LOCATION: location 48 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_R7.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - background 11 | - brith place 12 | - state 13 | - membership of a nation 14 | - habitancy 15 | - consultant 16 | - vicinity 17 | - business firm 18 | - seat of government 19 | - originator 20 | - persons in general 21 | - corporation 22 | delete_list: 23 | mapper: 24 | /business/company/advisors: consultant 25 | /business/company/founders: originator 26 | /business/company/industry: industry 27 | /business/company/major_shareholders: major shareholders 28 | /business/company/place_founded: place founded 29 | /business/company_shareholder/major_shareholder_of: major shareholder of 30 | /business/person/company: business firm 31 | /location/administrative_division/country: state 32 | /location/country/administrative_divisions: administrative divisions 33 | /location/country/capital: seat of government 34 | /location/location/contains: contains 35 | /location/neighborhood/neighborhood_of: vicinity 36 | /people/deceased_person/place_of_death: place of death 37 | /people/ethnicity/geographic_distribution: geographic distribution 38 | /people/ethnicity/people: persons in general 39 | /people/person/children: children 40 | /people/person/ethnicity: background 41 | /people/person/nationality: membership of a nation 42 | /people/person/place_lived: habitancy 43 | /people/person/place_of_birth: brith place 44 | /people/person/profession: profession 45 | /people/person/religion: religion 46 | /sports/sports_team/location: location 47 | /sports/sports_team_location/teams: teams 48 | ORGANIZATION: organization 49 | LOCATION: location 50 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V1.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | delete_list: 11 | mapper: 12 | /business/company/advisors: advisors 13 | /business/company/founders: business 14 | /business/company/industry: industry 15 | /business/company/major_shareholders: major shareholders 16 | /business/company/place_founded: place founded 17 | /business/company_shareholder/major_shareholder_of: business 18 | /business/person/company: business 19 | /location/administrative_division/country: location 20 | /location/country/administrative_divisions: location 21 | /location/country/capital: location 22 | /location/location/contains: contains 23 | /location/neighborhood/neighborhood_of: location 24 | /people/deceased_person/place_of_death: people 25 | /people/ethnicity/geographic_distribution: geographic distribution 26 | /people/ethnicity/people: people group 27 | /people/person/children: children 28 | /people/person/ethnicity: people 29 | /people/person/nationality: people 30 | /people/person/place_lived: people 31 | /people/person/place_of_birth: people 32 | /people/person/profession: profession 33 | /people/person/religion: religion 34 | /sports/sports_team/location: location place 35 | /sports/sports_team_location/teams: teams 36 | ORGANIZATION: organization 37 | LOCATION: location 38 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V2.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - capital 11 | - founders 12 | delete_list: 13 | mapper: 14 | /business/company/advisors: advisors 15 | /business/company/founders: founders 16 | /business/company/industry: industry 17 | /business/company/major_shareholders: major shareholders 18 | /business/company/place_founded: place founded 19 | /business/company_shareholder/major_shareholder_of: business 20 | /business/person/company: business 21 | /location/administrative_division/country: location 22 | /location/country/administrative_divisions: location 23 | /location/country/capital: capital 24 | /location/location/contains: contains 25 | /location/neighborhood/neighborhood_of: location 26 | /people/deceased_person/place_of_death: people 27 | /people/ethnicity/geographic_distribution: geographic distribution 28 | /people/ethnicity/people: people group 29 | /people/person/children: children 30 | /people/person/ethnicity: people 31 | /people/person/nationality: people 32 | /people/person/place_lived: people 33 | /people/person/place_of_birth: people 34 | /people/person/profession: profession 35 | /people/person/religion: religion 36 | /sports/sports_team/location: location place 37 | /sports/sports_team_location/teams: teams 38 | ORGANIZATION: organization 39 | LOCATION: location 40 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V3.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - capital 12 | - founders 13 | - place of death 14 | delete_list: 15 | mapper: 16 | /business/company/advisors: advisors 17 | /business/company/founders: founders 18 | /business/company/industry: industry 19 | /business/company/major_shareholders: major shareholders 20 | /business/company/place_founded: place founded 21 | /business/company_shareholder/major_shareholder_of: business 22 | /business/person/company: business 23 | /location/administrative_division/country: location 24 | /location/country/administrative_divisions: administrative divisions 25 | /location/country/capital: capital 26 | /location/location/contains: contains 27 | /location/neighborhood/neighborhood_of: location 28 | /people/deceased_person/place_of_death: place of death 29 | /people/ethnicity/geographic_distribution: geographic distribution 30 | /people/ethnicity/people: people group 31 | /people/person/children: children 32 | /people/person/ethnicity: people 33 | /people/person/nationality: people 34 | /people/person/place_lived: people 35 | /people/person/place_of_birth: people 36 | /people/person/profession: profession 37 | /people/person/religion: religion 38 | /sports/sports_team/location: location place 39 | /sports/sports_team_location/teams: teams 40 | ORGANIZATION: organization 41 | LOCATION: location 42 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V4.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - capital 12 | - company 13 | - founders 14 | - major shareholder of 15 | - place of death 16 | delete_list: 17 | mapper: 18 | /business/company/advisors: advisors 19 | /business/company/founders: founders 20 | /business/company/industry: industry 21 | /business/company/major_shareholders: major shareholders 22 | /business/company/place_founded: place founded 23 | /business/company_shareholder/major_shareholder_of: major shareholder of 24 | /business/person/company: company 25 | /location/administrative_division/country: location 26 | /location/country/administrative_divisions: administrative divisions 27 | /location/country/capital: capital 28 | /location/location/contains: contains 29 | /location/neighborhood/neighborhood_of: location 30 | /people/deceased_person/place_of_death: place of death 31 | /people/ethnicity/geographic_distribution: geographic distribution 32 | /people/ethnicity/people: people group 33 | /people/person/children: children 34 | /people/person/ethnicity: people 35 | /people/person/nationality: people 36 | /people/person/place_lived: people 37 | /people/person/place_of_birth: people 38 | /people/person/profession: profession 39 | /people/person/religion: religion 40 | /sports/sports_team/location: location place 41 | /sports/sports_team_location/teams: teams 42 | ORGANIZATION: organization 43 | LOCATION: location 44 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V5.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - capital 12 | - company 13 | - ethnicity 14 | - founders 15 | - major shareholder of 16 | - nationality 17 | - place of death 18 | delete_list: 19 | mapper: 20 | /business/company/advisors: advisors 21 | /business/company/founders: founders 22 | /business/company/industry: industry 23 | /business/company/major_shareholders: major shareholders 24 | /business/company/place_founded: place founded 25 | /business/company_shareholder/major_shareholder_of: major shareholder of 26 | /business/person/company: company 27 | /location/administrative_division/country: location 28 | /location/country/administrative_divisions: administrative divisions 29 | /location/country/capital: capital 30 | /location/location/contains: contains 31 | /location/neighborhood/neighborhood_of: location 32 | /people/deceased_person/place_of_death: place of death 33 | /people/ethnicity/geographic_distribution: geographic distribution 34 | /people/ethnicity/people: people group 35 | /people/person/children: children 36 | /people/person/ethnicity: ethnicity 37 | /people/person/nationality: nationality 38 | /people/person/place_lived: people 39 | /people/person/place_of_birth: people 40 | /people/person/profession: profession 41 | /people/person/religion: religion 42 | /sports/sports_team/location: location place 43 | /sports/sports_team_location/teams: teams 44 | ORGANIZATION: organization 45 | LOCATION: location 46 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V6.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | delete_list: 10 | new_list: 11 | - administrative divisions 12 | - capital 13 | - company 14 | - country 15 | - ethnicity 16 | - founders 17 | - major shareholder of 18 | - nationality 19 | - neighborhood of 20 | - place of death 21 | mapper: 22 | /business/company/advisors: advisors 23 | /business/company/founders: founders 24 | /business/company/industry: industry 25 | /business/company/major_shareholders: major shareholders 26 | /business/company/place_founded: place founded 27 | /business/company_shareholder/major_shareholder_of: major shareholder of 28 | /business/person/company: company 29 | /location/administrative_division/country: country 30 | /location/country/administrative_divisions: administrative divisions 31 | /location/country/capital: capital 32 | /location/location/contains: contains 33 | /location/neighborhood/neighborhood_of: neighborhood of 34 | /people/deceased_person/place_of_death: place of death 35 | /people/ethnicity/geographic_distribution: geographic distribution 36 | /people/ethnicity/people: people group 37 | /people/person/children: children 38 | /people/person/ethnicity: ethnicity 39 | /people/person/nationality: nationality 40 | /people/person/place_lived: people 41 | /people/person/place_of_birth: people 42 | /people/person/profession: profession 43 | /people/person/religion: religion 44 | /sports/sports_team/location: location place 45 | /sports/sports_team_location/teams: teams 46 | ORGANIZATION: organization 47 | LOCATION: location 48 | PERSON: person -------------------------------------------------------------------------------- /config/data_config/relation/NYT_V7.yaml: -------------------------------------------------------------------------------- 1 | name: NYT 2 | path: data/NYT-multi 3 | data_class: JointER 4 | split: 5 | train: train.json 6 | val: dev.json 7 | test: test.json 8 | language: en 9 | new_list: 10 | - administrative divisions 11 | - birth place 12 | - capital 13 | - company 14 | - country 15 | - ethnicity 16 | - founders 17 | - major shareholder of 18 | - nationality 19 | - neighborhood of 20 | - place lived 21 | - place of death 22 | delete_list: 23 | mapper: 24 | /business/company/advisors: advisors 25 | /business/company/founders: founders 26 | /business/company/industry: industry 27 | /business/company/major_shareholders: major shareholders 28 | /business/company/place_founded: place founded 29 | /business/company_shareholder/major_shareholder_of: major shareholder of 30 | /business/person/company: company 31 | /location/administrative_division/country: country 32 | /location/country/administrative_divisions: administrative divisions 33 | /location/country/capital: capital 34 | /location/location/contains: contains 35 | /location/neighborhood/neighborhood_of: neighborhood of 36 | /people/deceased_person/place_of_death: place of death 37 | /people/ethnicity/geographic_distribution: geographic distribution 38 | /people/ethnicity/people: people group 39 | /people/person/children: children 40 | /people/person/ethnicity: ethnicity 41 | /people/person/nationality: nationality 42 | /people/person/place_lived: place lived 43 | /people/person/place_of_birth: birth place 44 | /people/person/profession: profession 45 | /people/person/religion: religion 46 | /sports/sports_team/location: location place 47 | /sports/sports_team_location/teams: teams 48 | ORGANIZATION: organization 49 | LOCATION: location 50 | PERSON: person -------------------------------------------------------------------------------- /config/offset_map/closest_offset_en.yaml: -------------------------------------------------------------------------------- 1 | map_strategy: "closest" 2 | de_duplicate: True 3 | span_to_token: "space" 4 | -------------------------------------------------------------------------------- /config/offset_map/first_offset_en.yaml: -------------------------------------------------------------------------------- 1 | map_strategy: "first" 2 | de_duplicate: True 3 | span_to_token: "space" 4 | -------------------------------------------------------------------------------- /config/offset_map/longer_first_offset_en.yaml: -------------------------------------------------------------------------------- 1 | map_strategy: "longer_first" 2 | de_duplicate: True 3 | span_to_token: "space" 4 | -------------------------------------------------------------------------------- /config/prompt_conf/Few-NERD.ini: -------------------------------------------------------------------------------- 1 | export k8s_gpu_cards=1 2 | export gpu_node=${k8s_gpu_cards} 3 | 4 | export eval_steps=0 5 | export save_steps=0 6 | export epoch=20 7 | export max_source_length=300 8 | export max_target_length=300 9 | export task_name=entity 10 | export prompt_len=80 11 | export prompt_dim=800 12 | 13 | export BATCH_SIZE="12" 14 | export LR_RATE="1e-4" 15 | export WARMUP_PROP="0.06" 16 | export LABEL_SMOOTHING="0" 17 | export NEGATIVE="-1" 18 | export NOISE="0.1" 19 | export map_config='config/offset_map/closest_offset_en.yaml' 20 | 21 | export eval_match_mode="set" -------------------------------------------------------------------------------- /config/prompt_conf/Few-NERD_H.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=15 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name="entity" 14 | 15 | export lr_scheduler="linear" 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=True 30 | export prompt_len=80 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/Few-NERD_M.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=15 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name="entity" 14 | 15 | export lr_scheduler="linear" 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=False 30 | export prompt_len=80 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/Few-NERD_V.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=15 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name="entity" 14 | 15 | export lr_scheduler="linear" 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=True 30 | export prompt_len=80 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/NYT.ini: -------------------------------------------------------------------------------- 1 | export k8s_gpu_cards=1 2 | export gpu_node=${k8s_gpu_cards} 3 | 4 | export eval_steps=0 5 | export save_steps=0 6 | export epoch=30 7 | export max_source_length=256 8 | export task_name=relation 9 | export prompt_len=50 10 | export prompt_dim=800 11 | 12 | export BATCH_SIZE="16" 13 | export LR_RATE="1e-4" 14 | export WARMUP_PROP="0.06" 15 | export LABEL_SMOOTHING="0" 16 | export NEGATIVE="-1" 17 | export NOISE="0.1" 18 | export map_config='config/offset_map/closest_offset_en.yaml' 19 | 20 | export eval_match_mode="set" # 在eval_extraction.py中有体现，默认normal 21 | -------------------------------------------------------------------------------- /config/prompt_conf/NYT_H.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=20 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name="relation" 14 | 15 | export lr_scheduler="linear" 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=1 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=True 30 | export prompt_len=50 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/NYT_M.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=20 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name="relation" 14 | 15 | export lr_scheduler="linear" 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=1 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=True 30 | export prompt_len=50 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/NYT_V.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=20 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name="relation" 14 | 15 | export lr_scheduler="linear" 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=True 30 | export prompt_len=50 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/ace05_event.ini: -------------------------------------------------------------------------------- 1 | export k8s_gpu_cards=2 2 | export gpu_node=${k8s_gpu_cards} 3 | 4 | export eval_steps=0 5 | export save_steps=0 6 | export epoch=50 7 | export max_source_length=256 8 | export task_name=event 9 | export prompt_len=80 10 | export prompt_dim=512 11 | 12 | export BATCH_SIZE="16" 13 | export LR_RATE="1e-4" 14 | export WARMUP_PROP="0.06" 15 | export LABEL_SMOOTHING="0" 16 | export NEGATIVE="-1" 17 | export NOISE="0.1" 18 | export map_config='config/offset_map/closest_offset_en.yaml' 19 | 20 | export eval_match_mode="set" 21 | -------------------------------------------------------------------------------- /config/prompt_conf/ace05_event_H.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=30 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name=event 14 | 15 | export lr_scheduler=linear 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=False 30 | export prompt_len=80 31 | export prompt_dim=512 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/ace05_event_M.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=30 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name=event 14 | 15 | export lr_scheduler=linear 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=False 30 | export prompt_len=80 31 | export prompt_dim=512 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/ace05_event_R.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=30 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name=event 14 | 15 | export lr_scheduler=linear 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=0.8 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=False 30 | export prompt_len=80 31 | export prompt_dim=512 32 | 33 | -------------------------------------------------------------------------------- /config/prompt_conf/ace05_event_V.ini: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="1" 2 | export decoding_format='spotasoc' 3 | export constraint_decoding='' 4 | export fp16='' 5 | export seed=42 6 | export ordered_prompt=True 7 | export disable_tqdm=True 8 | export verbose=False 9 | 10 | export epoch=30 11 | export batch_size=16 12 | export max_source_length=256 13 | export task_name=event 14 | 15 | export lr_scheduler=linear 16 | export lr_rate=1e-4 17 | export warmup_ratio=0.06 18 | export label_smoothing=0 19 | export negative=-1 20 | export positive=1 21 | export spot_noise=0.1 22 | export asoc_noise=0.1 23 | export negative_ratio=1 24 | export eval_match_mode="normal" 25 | export map_config="config/offset_map/closest_offset_en.yaml" 26 | 27 | export use_ssi=True 28 | export use_prompt=True 29 | export use_task=True 30 | export prompt_len=80 31 | export prompt_dim=800 32 | 33 | -------------------------------------------------------------------------------- /dataset_construct/.gitignore: -------------------------------------------------------------------------------- 1 | /data 2 | /lightning_logs 3 | /model 4 | /models 5 | /*log 6 | /thirdparty 7 | /tmp 8 | 9 | .lock 10 | # mac 11 | .DS_Store 12 | 13 | # env 14 | /.vscode 15 | /.idea 16 | **/__pycache__ 17 | *.pyc 18 | .pytest_cache 19 | 20 | # doc 21 | docs/build 22 | docs/.vscode 23 | docs/source/_build 24 | 25 | -------------------------------------------------------------------------------- /dataset_construct/scripts/process_nerd.py: -------------------------------------------------------------------------------- 1 | def main(): 2 | src_path = "data/Few-NERD/supervised" 3 | tgt_path = "data/Few-NERD" 4 | for split in ["train.txt", "dev.txt", "test.txt"]: 5 | with open(f"{src_path}/{split}", 'r') as reader: 6 | with open(f"{tgt_path}/{split}", 'w') as writer: 7 | for line in reader: 8 | if line.strip() != "": 9 | n_line = line.strip().split('\t') 10 | if n_line[-1] != "O": 11 | line = f"{n_line[0]}\tI-{n_line[-1]}\n" 12 | writer.write(line) 13 | 14 | 15 | if __name__ == "__main__": 16 | main() -------------------------------------------------------------------------------- /dataset_construct/scripts/run_data_convert.bash: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | # -*- coding:utf-8 -*- 3 | 4 | python data_convert.py --config "../config/data_config/event" --iter_num=7 --task="ace05_event" --mode="H" 5 | python data_convert.py --config "../config/data_config/event" --iter_num=7 --task="ace05_event" --mode="V" 6 | python data_convert.py --config "../config/data_config/event" --iter_num=7 --task="ace05_event" --mode="M" 7 | python data_convert.py --config "../config/data_config/event" --iter_num=7 --task="ace05_event" --mode="R" 8 | 9 | python data_convert.py --config "../config/data_config/relation" --iter_num=7 --task="NYT" --mode="H" 10 | python data_convert.py --config "../config/data_config/relation" --iter_num=7 --task="NYT" --mode="V" 11 | python data_convert.py --config "../config/data_config/relation" --iter_num=7 --task="NYT" --mode="M" 12 | python data_convert.py --config "../config/data_config/relation" --iter_num=7 --task="NYT" --mode="R" 13 | 14 | python data_convert.py --config "../config/data_config/entity" --iter_num=7 --task="Few-NERD" --mode="H" 15 | python data_convert.py --config "../config/data_config/entity" --iter_num=7 --task="Few-NERD" --mode="V" 16 | python data_convert.py --config "../config/data_config/entity" --iter_num=7 --task="Few-NERD" --mode="M" 17 | python data_convert.py --config "../config/data_config/entity" --iter_num=7 --task="Few-NERD" --mode="R" 18 | -------------------------------------------------------------------------------- /dataset_construct/universal_ie/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/AdaKGC/227b7396499e0744f2ece879ddab4ac6517fc129/dataset_construct/universal_ie/__init__.py -------------------------------------------------------------------------------- /dataset_construct/universal_ie/dataset.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | import yaml 4 | import os 5 | from typing import Dict, List 6 | 7 | from universal_ie.utils import label_format 8 | import universal_ie.task_format as task_format 9 | 10 | 11 | class Dataset: 12 | def __init__(self, name: str, path: str, data_class: task_format.TaskFormat, split_dict: Dict, language: str, mapper: Dict, delete_list: List, other: Dict = None) -> None: 13 | self.name = name 14 | self.path = path 15 | self.data_class = data_class 16 | self.split_dict = split_dict 17 | self.language = language 18 | self.mapper = mapper 19 | self.other = other 20 | self.delete_list = delete_list 21 | 22 | def load_dataset(self, logger_name): 23 | datasets = {} 24 | for split_name, filename in self.split_dict.items(): 25 | datasets[split_name] = self.data_class.load_from_file( 26 | filename=os.path.join(self.path, filename), 27 | language=self.language, 28 | delete_list=self.delete_list, 29 | m=self.mapper, 30 | logger_name=logger_name, 31 | **self.other, 32 | ) 33 | return datasets 34 | 35 | @staticmethod 36 | def load_yaml_file(yaml_file): 37 | dataset_config = yaml.load(open(yaml_file), Loader=yaml.FullLoader) 38 | if 'mapper' in dataset_config: 39 | mapper = dataset_config['mapper'] 40 | for key in mapper: 41 | mapper[key] = label_format(mapper[key]) 42 | else: 43 | print(f"{dataset_config['name']} without label mapper.") 44 | mapper = None 45 | 46 | return Dataset( 47 | name=dataset_config['name'], 48 | path=dataset_config['path'], 49 | data_class=getattr(task_format, dataset_config['data_class']), 50 | split_dict=dataset_config['split'], 51 | language=dataset_config['language'], 52 | mapper=mapper, 53 | delete_list=dataset_config['delete_list'] if dataset_config['delete_list'] is not None else [], 54 | other=dataset_config.get('other', {}), 55 | ) 56 | -------------------------------------------------------------------------------- /dataset_construct/universal_ie/logger.py: -------------------------------------------------------------------------------- 1 | import logging 2 | import os 3 | 4 | def init_logger(task_name): 5 | logger = logging.getLogger(task_name) 6 | logger.setLevel(logging.INFO) 7 | 8 | logfile = './log/' + task_name + '_logger.txt' 9 | os.makedirs('./log/', exist_ok=True) 10 | 11 | fh = logging.FileHandler(logfile, mode = 'a', encoding = 'utf-8') 12 | fh.setLevel(logging.INFO) 13 | 14 | ch = logging.StreamHandler() 15 | ch.setLevel(logging.INFO) 16 | 17 | formatter = logging.Formatter("%(asctime)s: %(message)s") 18 | fh.setFormatter(formatter) 19 | ch.setFormatter(formatter) 20 | 21 | logger.addHandler(fh) 22 | logger.addHandler(ch) 23 | 24 | return logger -------------------------------------------------------------------------------- /dataset_construct/universal_ie/record_schema.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | import json 4 | 5 | class RecordSchema: 6 | def __init__(self, type_list, role_list, type_role_dict): 7 | self.type_list = type_list 8 | self.role_list = role_list 9 | self.type_role_dict = type_role_dict 10 | 11 | @staticmethod 12 | def read_from_file(filename): 13 | lines = open(filename).readlines() 14 | type_list = json.loads(lines[0]) 15 | role_list = json.loads(lines[1]) 16 | type_role_dict = json.loads(lines[2]) 17 | return RecordSchema(type_list, role_list, type_role_dict) 18 | 19 | def write_to_file(self, filename): 20 | with open(filename, 'w') as output: 21 | output.write(json.dumps(self.type_list, ensure_ascii=False) + '\n') 22 | output.write(json.dumps(self.role_list, ensure_ascii=False) + '\n') 23 | output.write(json.dumps(self.type_role_dict, ensure_ascii=False) + '\n') 24 | 25 | @staticmethod 26 | def output_schema(record_role_map, filename: str): 27 | """导出 Schema 文件 28 | Args: 29 | filename (str): [description] 30 | """ 31 | record_list = list(record_role_map.keys()) 32 | role_set = set() 33 | for record in record_role_map: 34 | role_set.update(record_role_map[record]) 35 | record_role_map[record] = list(record_role_map[record]) 36 | role_list = list(role_set) 37 | 38 | record_schema = RecordSchema(type_list=record_list, 39 | role_list=role_list, 40 | type_role_dict=record_role_map) 41 | record_schema.write_to_file(filename) -------------------------------------------------------------------------------- /dataset_construct/universal_ie/task_format/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from universal_ie.task_format.task_format import TaskFormat 4 | from universal_ie.task_format.oneie import OneIEEvent 5 | from universal_ie.task_format.jointer import JointER 6 | from universal_ie.task_format.nerd import NERD 7 | 8 | -------------------------------------------------------------------------------- /dataset_construct/universal_ie/task_format/task_format.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | import abc 4 | class TaskFormat: 5 | __metaclass__ = abc.ABCMeta 6 | 7 | @abc.abstractmethod 8 | def __init__(self, language='en'): 9 | self.language = language 10 | 11 | @abc.abstractmethod 12 | def generate_instance(self): 13 | pass 14 | 15 | @staticmethod 16 | @abc.abstractmethod 17 | def load_from_file(filename, language='en', name = 'train'): 18 | pass 19 | -------------------------------------------------------------------------------- /dataset_construct/universal_ie/utils.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from typing import List 4 | import os 5 | import sys 6 | 7 | 8 | global_mislabel_log = set() 9 | 10 | 11 | def tokens_to_str(tokens: List[str], language: str = 'en') -> str: 12 | if language == 'en': 13 | return ' '.join(tokens) 14 | elif language == 'zh': 15 | return ''.join(tokens) 16 | else: 17 | raise NotImplementedError('Language %s not supported' % language) 18 | 19 | 20 | def label_format(s): 21 | import re 22 | 23 | def uncamelize(s): 24 | re_outer = re.compile(r'([^A-Z ])([A-Z])') 25 | re_inner = re.compile(r'\b[A-Z]+(?=[A-Z][a-z])') 26 | sub = re_inner.sub(r'\g<0> ', re_outer.sub(r'\1 \2', s)).lower() 27 | return sub 28 | 29 | def remove(s): 30 | return s.replace("_", " ").replace("-", " ").replace(".", " ") 31 | 32 | s = remove(uncamelize(s)).split() 33 | if len(s) > 1 and s[0] == s[1]: 34 | s = s[1:] 35 | return " ".join(s) 36 | 37 | 38 | def load_dict_ini_file(filename): 39 | print("Warning: `load_dict_ini_file` is deprecated.") 40 | if not os.path.exists(filename): 41 | sys.stderr.write(f'[warning] cannot load label mapper from {filename}\n') 42 | return {} 43 | mapper = dict() 44 | for line in open(filename): 45 | key, value = line.strip().split('=') 46 | mapper[key] = label_format(value) 47 | return mapper 48 | 49 | 50 | def change_ptb_token_back(token): 51 | """将 PTBTokenized 的 Token 转换会原始字符串 52 | 53 | Args: 54 | token (str): PTBTokenize 后的 Token 字符串 55 | 56 | Returns: 57 | str: 原始 Token 字符串 58 | """ 59 | ptb_token_map = { 60 | '``': '"', 61 | "''": '"', 62 | '-LRB-': '(', 63 | '-RRB-': ')', 64 | '-LSB-': '[', 65 | '-RSB-': ']', 66 | '-LCB-': '{', 67 | '-RCB-': '}', 68 | } 69 | for ptb_token, raw_token in ptb_token_map.items(): 70 | if token == ptb_token: 71 | return raw_token 72 | return token 73 | 74 | 75 | def change_name_using_label_mapper(label_name, label_mapper): 76 | if label_mapper is None or len(label_mapper) == 0: 77 | return label_name 78 | if label_name not in label_mapper: 79 | print(f"{label_name} not found in mapper") 80 | global global_mislabel_log 81 | if label_name not in global_mislabel_log: 82 | global_mislabel_log.add(label_name) 83 | return label_mapper.get(label_name, label_name) 84 | 85 | 86 | 87 | 88 | 89 | 90 | 91 | -------------------------------------------------------------------------------- /etc/record.dataload.schema: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/AdaKGC/227b7396499e0744f2ece879ddab4ac6517fc129/etc/record.dataload.schema -------------------------------------------------------------------------------- /scripts/run_noprompt.bash: -------------------------------------------------------------------------------- 1 | mode=H 2 | data_name=ace05_event 3 | task=event 4 | device=4 5 | ratio=0.8 6 | bash scripts/fine_prompt.bash --model=hf_models/mix --data=data/${data_name}/${data_name}_${mode}/iter_1 --output=output/${data_name}_${mode}_${ratio} --config=config/prompt_conf/${data_name}.ini --device=${device} --negative_ratio=${ratio} --record2=data/${data_name}_${mode}/iter_7/record.schema 7 | python3 inference_mul.py --data=data/${data_name}/${data_name}_${mode} --model=output/${data_name}_${mode}_${ratio} --task=${task} --cuda=${device} --mode=${mode} 8 | -------------------------------------------------------------------------------- /scripts/run_prompt.bash: -------------------------------------------------------------------------------- 1 | mode=H 2 | data_name=ace05_event 3 | task=event 4 | device=4 5 | ratio=0.8 6 | bash scripts/run_prompt.bash --model=hf_models/mix --data=data/${data_name}/${data_name}_${mode}/iter_1 --output=output/${data_name}_${mode}_${ratio} --config=config/prompt_conf/${data_name}.ini --device=${device} --negative_ratio=${ratio} --record2=data/${data_name}_${mode}/iter_7/record.schema --use_prompt=True --init_prompt=True 7 | python3 inference_mul.py --data=data/${data_name}/${data_name}_${mode} --model=output/${data_name}_${mode}_${ratio} --task=${task} --cuda=${device} --mode=${mode} --use_prompt --use_ssi --prompt_len=80 --prompt_dim=512 8 | 9 | -------------------------------------------------------------------------------- /uie/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /uie/extraction/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /uie/extraction/constants.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | 4 | spot_prompt = '' 5 | asoc_prompt = '' 6 | 7 | type_start = '' 8 | type_end = '' 9 | text_start = '' 10 | span_start = '' 11 | null_span = '' 12 | null_label = '' 13 | 14 | spot_token = '' 15 | asoc_token = '' 16 | 17 | class StructureMarker: 18 | def __init__(self) -> None: 19 | pass 20 | 21 | 22 | class BaseStructureMarker(StructureMarker): 23 | def __init__(self) -> None: 24 | super().__init__() 25 | self.sent_start = '' 26 | self.sent_end = '' 27 | self.record_start = '' 28 | self.record_end = '' 29 | self.span_start = '' 30 | self.span_end = '' 31 | self.text_start = '' 32 | self.source_span_start = '' 33 | self.source_span_end = '' 34 | self.target_span_start = '' 35 | self.null_span = '' 36 | self.null_label = '' 37 | -------------------------------------------------------------------------------- /uie/extraction/extraction_metrics.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from typing import List 4 | from uie.extraction.record_schema import RecordSchema 5 | from uie.extraction.predict_parser import get_predict_parser, PredictParser 6 | from uie.extraction.scorer import Metric, RecordMetric, OrderedRecordMetric 7 | 8 | 9 | def eval_pred(predict_parser: PredictParser, gold_list, pred_list, text_list=None, raw_list=None): 10 | well_formed_list, counter = predict_parser.decode( 11 | gold_list, pred_list, text_list, raw_list 12 | ) 13 | 14 | spot_metric = Metric() 15 | asoc_metric = Metric() 16 | record_metric = RecordMetric() 17 | ordered_record_metric = OrderedRecordMetric() 18 | 19 | for instance in well_formed_list: 20 | spot_metric.count_instance(instance['gold_spot'], instance['pred_spot']) 21 | asoc_metric.count_instance(instance['gold_asoc'], instance['pred_asoc']) 22 | record_metric.count_instance(instance['gold_record'], instance['pred_record']) 23 | ordered_record_metric.count_instance(instance['gold_record'], instance['pred_record']) 24 | 25 | spot_result = spot_metric.compute_f1(prefix='spot-') 26 | asoc_result = asoc_metric.compute_f1(prefix='asoc-') 27 | record_result = record_metric.compute_f1(prefix='record-') 28 | ordered_record_result = ordered_record_metric.compute_f1(prefix='ordered-record-') 29 | 30 | overall_f1 = spot_result.get('spot-F1', 0.) + asoc_result.get('asoc-F1', 0.) 31 | # print(counter) 32 | result = {'overall-F1': overall_f1} 33 | result.update(spot_result) 34 | result.update(asoc_result) 35 | result.update(record_result) 36 | result.update(ordered_record_result) 37 | result.update(counter) 38 | return result 39 | 40 | 41 | def get_extract_metrics(pred_lns: List[str], tgt_lns: List[str], label_constraint: RecordSchema, decoding_format='tree'): 42 | predict_parser = get_predict_parser(decoding_schema=decoding_format, label_constraint=label_constraint) 43 | return eval_pred( 44 | predict_parser=predict_parser, 45 | gold_list=tgt_lns, 46 | pred_list=pred_lns 47 | ) 48 | -------------------------------------------------------------------------------- /uie/extraction/label_tree.py: -------------------------------------------------------------------------------- 1 | from typing import Dict 2 | 3 | 4 | def list_dictionary(d, n_tab=-1): 5 | if isinstance(d, list): 6 | for i in d: 7 | list_dictionary(i, n_tab) 8 | elif isinstance(d, dict): 9 | n_tab += 1 10 | for key, value in d.items(): 11 | if key == '': 12 | print("{}{}".format(" " * n_tab, key)) 13 | else: 14 | print("{}{}".format(" " * n_tab, key)) 15 | list_dictionary(value, n_tab) 16 | else: 17 | print("{}{}".format("\t" * n_tab, d)) 18 | 19 | 20 | def print_tree(tree): 21 | list_dictionary(tree) 22 | 23 | 24 | def get_label_name_tree(label_name_list, tokenizer, end_symbol=''): 25 | ''' 26 | label_name_list: 27 | ["end position", "attack", "start position", "nominate", "charge indict", "transfer money", "transfer ownership", "release parole", "born", "sentence", "die", "demonstrate", "arrest jail", "transport", "elect", "start organization", "meet", "injure", "phone write", "merge organization", "declare bankruptcy", "trial hearing", "fine", "convict", "end organization", "sue", "divorce", "acquit", "appeal", "marry", "execute", "extradite", "pardon"] 28 | ["person", "place", "destination", "giver", "plaintiff", "instrument", "attacker", "agent", "origin", "victim", "seller", "vehicle", "buyer", "organization", "adjudicator", "defendant", "beneficiary", "artifact", "prosecutor", "recipient", "target", "entity"] 29 | ''' 30 | 31 | sub_token_tree = dict() 32 | 33 | label_tree = dict() # {"end position":[], "attack":[],...} 34 | for typename in label_name_list: 35 | after_tokenized = tokenizer.encode(typename, add_special_tokens=False) 36 | # label_tree[typename] = tokenizer.convert_ids_to_tokens(after_tokenized) 37 | label_tree[typename] = after_tokenized 38 | 39 | for _, sub_label_seq in label_tree.items(): 40 | parent = sub_token_tree 41 | for value in sub_label_seq: 42 | if value not in parent: 43 | parent[value] = dict() 44 | parent = parent[value] 45 | 46 | parent[end_symbol] = None 47 | 48 | return sub_token_tree 49 | 50 | 51 | def get_type_role_tree(type_role_dict, role_tree, tokenizer, end_symbol=''): 52 | type_role_tree = dict() 53 | for typename, role_list in type_role_dict.items(): 54 | typetoken = tokenizer.encode(typename, add_special_tokens=False) 55 | sub_token_tree = dict() 56 | for role in role_list: 57 | role_token = tokenizer.encode(role, add_special_tokens=False) 58 | sub_token_tree.update({role_token[0]: role_tree[role_token[0]]}) 59 | type_role_tree[tuple(typetoken)] = sub_token_tree 60 | return type_role_tree 61 | 62 | 63 | 64 | class PrefixTree: 65 | def __init__(self, label_name_list, tokenizer, end_symbol=''): 66 | self.label_name_list = label_name_list 67 | self._tokenizer = tokenizer 68 | self.label_name_tree = get_label_name_tree(label_name_list, tokenizer, end_symbol) 69 | self._end_symbol = end_symbol 70 | 71 | def is_end_of_tree(self, tree: Dict): 72 | return len(tree) == 1 and self._end_symbol in tree 73 | -------------------------------------------------------------------------------- /uie/extraction/noiser/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /uie/extraction/predict_parser/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from uie.extraction.predict_parser.predict_parser import PredictParser 4 | from uie.extraction.predict_parser.spotasoc_predict_parser import SpotAsocPredictParser 5 | 6 | 7 | decoding_format_dict = { 8 | 'spotasoc': SpotAsocPredictParser, 9 | } 10 | 11 | 12 | def get_predict_parser(decoding_schema, label_constraint): 13 | return decoding_format_dict[decoding_schema](label_constraint=label_constraint) 14 | -------------------------------------------------------------------------------- /uie/extraction/predict_parser/predict_parser.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from typing import List, Counter, Tuple 4 | 5 | 6 | class PredictParser: 7 | def __init__(self, label_constraint=None): 8 | self.spot_set = label_constraint.type_list if label_constraint else list() 9 | self.role_set = label_constraint.role_list if label_constraint else list() 10 | 11 | def decode(self, gold_list, pred_list, text_list=None, raw_list=None) -> Tuple[List, Counter]: 12 | pass 13 | -------------------------------------------------------------------------------- /uie/extraction/record_schema.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | import json 4 | from collections import defaultdict 5 | from typing import List 6 | 7 | 8 | class RecordSchema: 9 | def __init__(self, type_list, role_list, type_role_dict): 10 | self.type_list = type_list 11 | self.role_list = role_list 12 | self.type_role_dict = type_role_dict 13 | 14 | def __repr__(self) -> str: 15 | return f"Type: {self.type_list}\n" \ 16 | f"Role: {self.role_list}\n" \ 17 | f"Map: {self.type_role_dict}" 18 | 19 | @staticmethod 20 | def get_empty_schema(): 21 | return RecordSchema(type_list=list(), role_list=list(), type_role_dict=dict()) 22 | 23 | @staticmethod 24 | def read_from_file(filename): 25 | lines = open(filename).readlines() 26 | type_list = json.loads(lines[0]) 27 | role_list = json.loads(lines[1]) 28 | type_role_dict = json.loads(lines[2]) 29 | return RecordSchema(type_list, role_list, type_role_dict) 30 | 31 | def write_to_file(self, filename): 32 | with open(filename, 'w') as output: 33 | output.write(json.dumps(self.type_list) + '\n') 34 | output.write(json.dumps(self.role_list) + '\n') 35 | output.write(json.dumps(self.type_role_dict) + '\n') 36 | 37 | 38 | def merge_schema(schema_list: List[RecordSchema]): 39 | type_set = set() 40 | role_set = set() 41 | type_role_dict = defaultdict(list) 42 | 43 | for schema in schema_list: 44 | 45 | for type_name in schema.type_list: 46 | type_set.add(type_name) 47 | 48 | for role_name in schema.role_list: 49 | role_set.add(role_name) 50 | 51 | for type_name in schema.type_role_dict: 52 | type_role_dict[type_name] += schema.type_role_dict[type_name] 53 | 54 | for type_name in type_role_dict: 55 | type_role_dict[type_name] = list(set(type_role_dict[type_name])) 56 | 57 | return RecordSchema(type_list=list(type_set), 58 | role_list=list(role_set), 59 | type_role_dict=type_role_dict, 60 | ) 61 | -------------------------------------------------------------------------------- /uie/extraction/utils.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | 4 | 5 | def convert_spot_asoc(spot_asoc_instance, structure_maker, null_text = ""): 6 | ''' 7 | 将 spot asoc 结构转换成SEL表达式 8 | ''' 9 | spot_instance_str_rep_list = list() 10 | for spot in spot_asoc_instance: 11 | spot_str_rep = [ 12 | spot['label'], 13 | structure_maker.target_span_start, 14 | spot['span'], 15 | ] 16 | for asoc_label, asoc_span in spot.get('asoc', list()): 17 | asoc_str_rep = [ 18 | structure_maker.span_start, 19 | asoc_label, 20 | structure_maker.target_span_start, 21 | asoc_span, 22 | structure_maker.span_end, 23 | ] 24 | spot_str_rep += [' '.join(asoc_str_rep)] 25 | spot_instance_str_rep_list += [' '.join([ 26 | structure_maker.record_start, 27 | ' '.join(spot_str_rep), 28 | structure_maker.record_end, 29 | ])] 30 | if null_text != "": 31 | spot_instance_str_rep_list += [null_text] 32 | target_text = ' '.join([ 33 | structure_maker.sent_start, 34 | ' '.join(spot_instance_str_rep_list), 35 | structure_maker.sent_end, 36 | ]) 37 | return target_text 38 | 39 | 40 | def convert_spot_asoc_name(spot_asoc_instance, structure_maker): 41 | """将一个 Spot-Asoc-Name 实例转换成目标字符串 42 | 43 | Args: 44 | spot_asoc_instance ([type]): [description] 45 | structure_maker ([type]): [description] 46 | 47 | Returns: 48 | [type]: [description] 49 | """ 50 | spot_instance_str_rep_list = list() 51 | for spot in spot_asoc_instance: 52 | spot_str_rep = [ 53 | spot['span'], 54 | structure_maker.target_span_start, 55 | spot['label'], 56 | ] 57 | for asoc_label, asoc_span in spot.get('asoc', list()): 58 | asoc_str_rep = [ 59 | structure_maker.span_start, 60 | asoc_span, 61 | structure_maker.target_span_start, 62 | asoc_label, 63 | structure_maker.span_end, 64 | ] 65 | spot_str_rep += [' '.join(asoc_str_rep)] 66 | spot_instance_str_rep_list += [' '.join([ 67 | structure_maker.record_start, 68 | ' '.join(spot_str_rep), 69 | structure_maker.record_end, 70 | ])] 71 | target_text = ' '.join([ 72 | structure_maker.sent_start, 73 | ' '.join(spot_instance_str_rep_list), 74 | structure_maker.sent_end, 75 | ]) 76 | return target_text 77 | 78 | 79 | convert_to_record_function = { 80 | 'spotasoc': convert_spot_asoc, 81 | 'spotasocname': convert_spot_asoc_name, 82 | } 83 | -------------------------------------------------------------------------------- /uie/sel2record/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/AdaKGC/227b7396499e0744f2ece879ddab4ac6517fc129/uie/sel2record/__init__.py -------------------------------------------------------------------------------- /uie/seq2seq/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | -------------------------------------------------------------------------------- /uie/seq2seq/constraint_decoder/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from uie.seq2seq.constraint_decoder.spotasoc_constraint_decoder import ( 4 | SpotAsocConstraintDecoder, 5 | SpotConstraintDecoder 6 | ) 7 | 8 | 9 | def get_constraint_decoder(tokenizer, type_schema, decoding_schema, task_name='event', source_prefix=None): 10 | if decoding_schema == 'spotasoc': 11 | if len(type_schema.role_list) == 0: 12 | task_map = { 13 | 'entity': SpotConstraintDecoder, 14 | 'relation': SpotConstraintDecoder, 15 | 'event': SpotConstraintDecoder, 16 | 'record': SpotConstraintDecoder, 17 | } 18 | else: 19 | task_map = { 20 | 'entity': SpotAsocConstraintDecoder, 21 | 'relation': SpotAsocConstraintDecoder, 22 | 'event': SpotAsocConstraintDecoder, 23 | 'record': SpotAsocConstraintDecoder, 24 | } 25 | else: 26 | raise NotImplementedError( 27 | f'Type Schema {type_schema}, Decoding Schema {decoding_schema}, Task {task_name} do not map to constraint decoder.' 28 | ) 29 | return task_map[task_name](tokenizer=tokenizer, type_schema=type_schema, source_prefix=source_prefix) 30 | -------------------------------------------------------------------------------- /uie/seq2seq/constraint_decoder/constraint_decoder.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | from collections import defaultdict 4 | import os 5 | from typing import List 6 | 7 | 8 | def match_sublist(the_list, to_match): # the_list: input+text, to_match 9 | """ 10 | 11 | :param the_list: [1, 2, 3, 4, 5, 6, 1, 2, 4, 5] 12 | :param to_match: 13 | [1, 2] 14 | :return: 15 | [(0, 1), (6, 7)] 16 | """ 17 | len_to_match = len(to_match) 18 | matched_list = list() 19 | for index in range(len(the_list) - len_to_match + 1): 20 | if to_match == the_list[index:index + len_to_match]: 21 | matched_list += [(index, index + len_to_match - 1)] 22 | return matched_list 23 | 24 | 25 | def find_bracket_position(generated_text, _type_start, _type_end): 26 | bracket_position = {_type_start: list(), _type_end: list()} 27 | for index, char in enumerate(generated_text): 28 | if char in bracket_position: 29 | bracket_position[char] += [index] 30 | return bracket_position 31 | 32 | 33 | def build_sentence_tree(sentence): 34 | tree = defaultdict(set) 35 | 36 | for prev_token, next_token in zip(sentence[:-1], sentence[1:]): 37 | tree[prev_token].add(next_token) 38 | 39 | return tree 40 | 41 | 42 | def generated_search_prefix_tree(generated, prefix_tree, tokenizer): 43 | tree = prefix_tree 44 | # Leaf is KEY_VALUE_SPLIT 45 | for token in generated: 46 | 47 | if token not in tree: 48 | return [tokenizer.eos_token] 49 | tree = tree[token] 50 | 51 | return list(tree) 52 | 53 | 54 | def generated_search_src_sequence(generated, src_sentence, end_sequence_search_tokens=None): 55 | 56 | if len(generated) == 0: # 已生成的text token为空 57 | # All src tokens are valid before generation 58 | return src_sentence 59 | 60 | matched_tuples = match_sublist(the_list=src_sentence, to_match=generated) # 已生成的text token>=1 61 | 62 | valid_token = list() 63 | for _, end in matched_tuples: 64 | next_index = end + 1 65 | if next_index < len(src_sentence): 66 | valid_token += [src_sentence[next_index]] 67 | 68 | if end_sequence_search_tokens: 69 | valid_token += end_sequence_search_tokens # 后一个token++ 70 | 71 | return valid_token 72 | 73 | 74 | class ConstraintDecoder: 75 | def __init__(self, tokenizer, source_prefix): 76 | self.tokenizer = tokenizer 77 | self.source_prefix = source_prefix 78 | self.source_prefix_tokenized = tokenizer.encode(source_prefix, 79 | add_special_tokens=False) if source_prefix else [] 80 | 81 | def get_state_valid_tokens(self, src_sentence: List[str], tgt_generated: List[str]) -> List[str]: 82 | pass 83 | 84 | def constraint_decoding(self, src_sentence, tgt_generated): 85 | if self.source_prefix_tokenized: 86 | # Remove Source Prefix for Generation 87 | src_sentence = src_sentence[len(self.source_prefix_tokenized):] 88 | 89 | valid_token_ids = self.get_state_valid_tokens(src_sentence.tolist(), tgt_generated.tolist()) 90 | 91 | return valid_token_ids 92 | -------------------------------------------------------------------------------- /uie/seq2seq/data_collator/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # -*- coding:utf-8 -*- 3 | 4 | 5 | from uie.seq2seq.data_collator.meta_data_collator import ( 6 | DataCollatorForMetaSeq2Seq, 7 | DynamicSSIGenerator, 8 | PromptForMetaSeq2Seq, 9 | PromptSSIGenerator, 10 | ) 11 | 12 | 13 | 14 | __all__ = [ 15 | 'DataCollatorForMetaSeq2Seq', 16 | 'DynamicSSIGenerator', 17 | 'HybirdDataCollator', 18 | 'DataCollatorForT5MLM', 19 | 'PromptForMetaSeq2Seq', 20 | 'PromptSSIGenerator', 21 | ] 22 | --------------------------------------------------------------------------------