Untitled.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "364275aa-456e-4700-a14d-863a0851d209",
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "\n",
    "df = pd.read_json('ASDiv_instruct.json')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "8c687a80-ea4b-4c83-8a89-7a4a319e1acf",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>instruction</th>\n",
       "      <th>input</th>\n",
       "      <th>output</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>Write a grade 1 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Seven red apples and two green apples are in t...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>Write a grade 1 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Ellen has six more balls than Marin. Marin has...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>Write a grade 1 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Janet has nine oranges and Sharon has seven or...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>Write a grade 1 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Allan brought two balloons and Jake brought fo...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>Write a grade 1 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Adam has five more apples than Jackie. Jackie ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2300</th>\n",
       "      <td>Write a grade 4 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>There are 3300 bananas in Janice's banana coll...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2301</th>\n",
       "      <td>Write a grade 4 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Janice has 316 bottle caps that must be put aw...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2302</th>\n",
       "      <td>Write a grade 4 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Chris is inviting 82 friends to a party. He ha...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2303</th>\n",
       "      <td>Write a grade 4 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>Keith has 5530 marbles and 3 pencils. If he sh...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2304</th>\n",
       "      <td>Write a grade 4 math word problem.</td>\n",
       "      <td></td>\n",
       "      <td>There are 43 students and 1720 apples. Each st...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>2305 rows × 3 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                             instruction input  \\\n",
       "0     Write a grade 1 math word problem.         \n",
       "1     Write a grade 1 math word problem.         \n",
       "2     Write a grade 1 math word problem.         \n",
       "3     Write a grade 1 math word problem.         \n",
       "4     Write a grade 1 math word problem.         \n",
       "...                                  ...   ...   \n",
       "2300  Write a grade 4 math word problem.         \n",
       "2301  Write a grade 4 math word problem.         \n",
       "2302  Write a grade 4 math word problem.         \n",
       "2303  Write a grade 4 math word problem.         \n",
       "2304  Write a grade 4 math word problem.         \n",
       "\n",
       "                                                 output  \n",
       "0     Seven red apples and two green apples are in t...  \n",
       "1     Ellen has six more balls than Marin. Marin has...  \n",
       "2     Janet has nine oranges and Sharon has seven or...  \n",
       "3     Allan brought two balloons and Jake brought fo...  \n",
       "4     Adam has five more apples than Jackie. Jackie ...  \n",
       "...                                                 ...  \n",
       "2300  There are 3300 bananas in Janice's banana coll...  \n",
       "2301  Janice has 316 bottle caps that must be put aw...  \n",
       "2302  Chris is inviting 82 friends to a party. He ha...  \n",
       "2303  Keith has 5530 marbles and 3 pencils. If he sh...  \n",
       "2304  There are 43 students and 1720 apples. Each st...  \n",
       "\n",
       "[2305 rows x 3 columns]"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "14188271-8472-44c3-86f3-1c32890bee92",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: Elliott has 8 candies and Sabetha has 10 candies. Find the ratio of Elliott's candies to Sabetha's candies.\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: One number is half the other. Sum of two numbers is equal to 6. Find the numbers.\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: Victor was playing checkers with a friend. The ratio of games Victor won was 9:5. If Victor won 36 games, how many games did his friend win?\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: Pete, Bryan and Philip are cousins. Pete's age is one-third of Bryan and Philip is five years elder than Bryan. If the sum of the age of the cousins is 40, find the ages of each.\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: Fred bought five new baseball trading cards to add to his collection. The next day his dog ate half of his collection. There are now only twenty-six cards left. How many cards did Fred start with ?\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: A sofa and a love seat together costs $444. The sofa costs double the love seat. How much do they each cost?\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: Kathy scored 360 in first quarter and 420 in second quarters. Find the ratio of scores in the first quarter to the second quarter.\n",
      "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n",
      "\n",
      "### Instruction:\n",
      "Write a grade 6 math word problem.\n",
      "\n",
      "### Response: \n"
     ]
    }
   ],
   "source": [
    "import random\n",
    "grades = ['1', '2', '3', '4', '5', '6']\n",
    "grade = random.choice(grades)\n",
    "prompt = f\"Write a grade {grade} math word problem.\"\n",
    "questions = []\n",
    "for i in range(0, 7):\n",
    "    question = df.query(f\"instruction=='{prompt}'\")['output'].iloc[random.randint(0,len(df.query(f\"instruction=='{prompt}'\")['instruction']))]\n",
    "    questions.append(question)\n",
    "formatted_prompt = []\n",
    "for i in range(0,7):\n",
    "    formatted_prompt.append((f\"Below is an instruction that describes a task. \"\n",
    "                f\"Write a response that appropriately completes the request.\\n\\n\"\n",
    "                f\"### Instruction:\\n{prompt}\\n\\n### Response: {questions[i]}\"))\n",
    "formatted_prompt.append(f\"Below is an instruction that describes a task. \"\n",
    "                f\"Write a response that appropriately completes the request.\\n\\n\"\n",
    "                f\"### Instruction:\\n{prompt}\\n\\n### Response: \")\n",
    "formatted_prompt = \"\\n\".join(formatted_prompt)\n",
    "print(formatted_prompt)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3c715fbb-d40e-49a7-b8c1-dcec2db5547e",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "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.8.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}