-
Notifications
You must be signed in to change notification settings - Fork 10
Open
Description
RAG4RE/src/data_augmentation/prompt_generation/prompt_generation.py
Lines 156 to 162 in a5f02f4
| train_data = read_json("/Users/sefika/phd_projects/revision/RAG4RE/data/semeval/original_data/train_sentences.json") | |
| relation_data = read_json("/Users/sefika/phd_projects/revision/RAG4RE/data/semeval/original_data/train_relations.json") | |
| relation_names = read_json("/Users/sefika/phd_projects/revision/RAG4RE/data/semeval/original_data/relations.json")['relation']['names'] | |
| train_relations = [relation_names[relation] for relation in relation_data] | |
| prompts = generate_prompts(train_data, train_relations, [], dataset="semeval", prompt_type="simple") | |
| write_json( "/Users/sefika/phd_projects/revision/RAG4RE/data/semeval/original_data/train_prompts.json", prompts) |
Metadata
Metadata
Assignees
Labels
No labels