aboutsummaryrefslogtreecommitdiff
path: root/gnqa/paper2_eval/src/create_dataset.py
diff options
context:
space:
mode:
authorShelbySolomonDarnell2024-10-17 12:24:26 +0300
committerShelbySolomonDarnell2024-10-17 12:24:26 +0300
commit00cba4b9a1e88891f1f96a1199320092c1962343 (patch)
tree270fd06daa18b2fc5687ee72d912cad771354bb0 /gnqa/paper2_eval/src/create_dataset.py
parente0b2b0e55049b89805f73f291df1e28fa05487fe (diff)
downloadgn-ai-master.tar.gz
Docker image built to run code, all evals run using R2RHEADmaster
Diffstat (limited to 'gnqa/paper2_eval/src/create_dataset.py')
-rw-r--r--gnqa/paper2_eval/src/create_dataset.py20
1 files changed, 20 insertions, 0 deletions
diff --git a/gnqa/paper2_eval/src/create_dataset.py b/gnqa/paper2_eval/src/create_dataset.py
new file mode 100644
index 0000000..5f0efa9
--- /dev/null
+++ b/gnqa/paper2_eval/src/create_dataset.py
@@ -0,0 +1,20 @@
+#!/usr/bin/python3
+
+import sys
+from document_operations import DocOps
+
+
+'''
+*******************************************************************************
+Commands
+*******************************************************************************
+'''
+
+try:
+ read_file = str(sys.argv[1])
+ out_file = str(sys.argv[2])
+except:
+ exit('Example use "python create_dataset.py ../data/lists/human_list_cs_gn.json ../data/dataset/human_cs_gn.json"')
+
+doc_list = DocOps.read_json_document(read_file)
+DocOps.combine_responses(doc_list, out_file) \ No newline at end of file