Spaces:
Running
Running
File size: 2,417 Bytes
5881f5e cd7c7a0 c878ef3 379ea31 5881f5e 379ea31 5881f5e 3ae2694 5881f5e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
{
"cells": [
{
"cell_type": "code",
"execution_count": null,
"id": "a5e875f5",
"metadata": {},
"outputs": [],
"source": [
"import pandas as pd\n",
"\n# Code that generates the output\n\n",
"answers = pd.DataFrame({\n",
" \"id\": [1, 2],\n",
" \"question\": [\"What is the capital of France?\", \"What is the capital of Germany?\"],\n",
" \"passages\": [\n",
" [\n",
" {\"passage\": \"Paris is the capital and most populous city of France.\", \n",
" \"doc_IDs\": [\"<urn:uuid:1234abcd-5678-efgh-9101-ijklmnopqrst>\", \"<urn:uuid:1234abcd-5678-efgh-9202-ijklmnopqrst>\"]},\n",
" {\"passage\": \"France is located in Western Europe.\", \n",
" \"doc_IDs\": [\"<urn:uuid:1234abcd-5678-efgh-9101-ijklmnopqrst>\"]}\n",
" ],\n",
" [\n",
" {\"passage\": \"Berlin is the capital of Germany.\", \n",
" \"doc_IDs\": [\"<urn:uuid:1234abcd-5678-efgh-9101-ijklmnopqrst>\"]}\n",
" ]\n",
" ],\n",
" \"final_prompt\": [\n",
" \"Using the following - Paris is the capital and most populous city of France - and - France is located in Western Europe - answer the question: What is the capital of France?\",\n",
" \"Using the following - Berlin is the capital of Germany - answer the question: What is the capital of Germany?\"\n",
" ],\n",
" \"answer\": [\"Paris\", \"Berlin\"]\n",
"})\n\n",
"# Convert to JSON format\n",
"answers_json = answers.to_json(orient='records', lines=True, force_ascii=False)\n",
"\n",
"# Or just save to a file\n",
"answers.to_json(\"answers.jsonl\", orient='records', lines=True, force_ascii=False)\n",
"\n",
"# Load the file to make sure it is correct\n",
"loaded_answers = pd.read_json(\"answers.jsonl\", lines=True)\n\n",
"# Print the loaded answers\n",
"print(loaded_answers)\n"
]
}
],
"metadata": {
"kernelspec": {
"display_name": ".venv",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.12.8"
}
},
"nbformat": 4,
"nbformat_minor": 5
}
|