{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "10676e1f-9a7d-453f-9334-246ebb2142c9", "metadata": {}, "outputs": [], "source": [ "datapath = 'readsy/stories/'\n", "pairpath = 'readsy/pairs/readsy_story_pairs0407.csv'\n", "from transformers import AutoTokenizer\n", "tokenizer = AutoTokenizer.from_pretrained(\"unsloth/gemma-2b-bnb-4bit\")\n" ] }, { "cell_type": "code", "execution_count": 2, "id": "7d73d0b5-7356-4c06-96bc-91dc807bcb0d", "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "the total number of pairs is 100\n", "the number of effective pairs is 91\n" ] }, { "name": "stderr", "output_type": "stream", "text": [ "No chat template is set for this tokenizer, falling back to a default class-level template. This is very error-prone, because models are often trained with templates different from the class default! Default chat templates are a legacy feature and will be removed in Transformers v4.43, at which point any code depending on them will stop working. We recommend setting a valid chat template before then to ensure that this model continues working without issues.\n" ] }, { "name": "stdout", "output_type": "stream", "text": [ "Index(['prompt_id', 'prompt', 'story_id', 'story_title', 'story_author',\n", " 'story_url', 'link', 'genre', 'is_sensitive', 'categories', 'likes',\n", " 'story_text', 'posted_date', 'comments'],\n", " dtype='object')\n", "the columns of train is Index(['prompt_id', 'story1_id', 'story2_id', 'time_lag', 'least_likes'], dtype='object')\n", "prompt_1234\n", "78559 Write a story about someone who finds somethin...\n", "78560 Write a story about someone who finds somethin...\n", "78561 Write a story about someone who finds somethin...\n", "78562 Write a story about someone who finds somethin...\n", "78563 Write a story about someone who finds somethin...\n", " ... \n", "78644 Write a story about someone who finds somethin...\n", "78645 Write a story about someone who finds somethin...\n", "78646 Write a story about someone who finds somethin...\n", "78647 Write a story about someone who finds somethin...\n", "78648 Write a story about someone who finds somethin...\n", "Name: prompt, Length: 90, dtype: object\n", "prompt_0852\n", "24702 End your story with a character standing in th...\n", "24703 End your story with a character standing in th...\n", "24704 End your story with a character standing in th...\n", "24705 End your story with a character standing in th...\n", "24706 End your story with a character standing in th...\n", " ... \n", "24799 End your story with a character standing in th...\n", "24800 End your story with a character standing in th...\n", "24801 End your story with a character standing in th...\n", "24802 End your story with a character standing in th...\n", "24803 End your story with a character standing in th...\n", "Name: prompt, Length: 102, dtype: object\n", "prompt_1153\n", "66372 Write a story about someone who's haunted by t...\n", "66373 Write a story about someone who's haunted by t...\n", "66374 Write a story about someone who's haunted by t...\n", "66375 Write a story about someone who's haunted by t...\n", "66376 Write a story about someone who's haunted by t...\n", " ... \n", "66621 Write a story about someone who's haunted by t...\n", "66622 Write a story about someone who's haunted by t...\n", "66623 Write a story about someone who's haunted by t...\n", "66624 Write a story about someone who's haunted by t...\n", "66625 Write a story about someone who's haunted by t...\n", "Name: prompt, Length: 254, dtype: object\n", "prompt_1167\n", "69390 Write a story that takes place in a waiting room.\n", "69391 Write a story that takes place in a waiting room.\n", "69392 Write a story that takes place in a waiting room.\n", "69393 Write a story that takes place in a waiting room.\n", "69394 Write a story that takes place in a waiting room.\n", " ... \n", "69643 Write a story that takes place in a waiting room.\n", "69644 Write a story that takes place in a waiting room.\n", "69645 Write a story that takes place in a waiting room.\n", "69646 Write a story that takes place in a waiting room.\n", "69647 Write a story that takes place in a waiting room.\n", "Name: prompt, Length: 258, dtype: object\n", "prompt_0901\n", "28473 Frame your story as an adult recalling the eve...\n", "28474 Frame your story as an adult recalling the eve...\n", "28475 Frame your story as an adult recalling the eve...\n", "28476 Frame your story as an adult recalling the eve...\n", "28477 Frame your story as an adult recalling the eve...\n", " ... \n", "28714 Frame your story as an adult recalling the eve...\n", "28715 Frame your story as an adult recalling the eve...\n", "28716 Frame your story as an adult recalling the eve...\n", "28717 Frame your story as an adult recalling the eve...\n", "28718 Frame your story as an adult recalling the eve...\n", "Name: prompt, Length: 246, dtype: object\n", "prompt_1160\n", "68088 Write a story about a person experiencing pre-...\n", "68089 Write a story about a person experiencing pre-...\n", "68090 Write a story about a person experiencing pre-...\n", "68091 Write a story about a person experiencing pre-...\n", "68092 Write a story about a person experiencing pre-...\n", " ... \n", "68245 Write a story about a person experiencing pre-...\n", "68246 Write a story about a person experiencing pre-...\n", "68247 Write a story about a person experiencing pre-...\n", "68248 Write a story about a person experiencing pre-...\n", "68249 Write a story about a person experiencing pre-...\n", "Name: prompt, Length: 162, dtype: object\n", "prompt_0712\n", "18628 Start your story with the narrator or a charac...\n", "18629 Start your story with the narrator or a charac...\n", "18630 Start your story with the narrator or a charac...\n", "18631 Start your story with the narrator or a charac...\n", "18632 Start your story with the narrator or a charac...\n", " ... \n", "18726 Start your story with the narrator or a charac...\n", "18727 Start your story with the narrator or a charac...\n", "18728 Start your story with the narrator or a charac...\n", "18729 Start your story with the narrator or a charac...\n", "18730 Start your story with the narrator or a charac...\n", "Name: prompt, Length: 103, dtype: object\n", "prompt_0901\n", "28473 Frame your story as an adult recalling the eve...\n", "28474 Frame your story as an adult recalling the eve...\n", "28475 Frame your story as an adult recalling the eve...\n", "28476 Frame your story as an adult recalling the eve...\n", "28477 Frame your story as an adult recalling the eve...\n", " ... \n", "28714 Frame your story as an adult recalling the eve...\n", "28715 Frame your story as an adult recalling the eve...\n", "28716 Frame your story as an adult recalling the eve...\n", "28717 Frame your story as an adult recalling the eve...\n", "28718 Frame your story as an adult recalling the eve...\n", "Name: prompt, Length: 246, dtype: object\n", "prompt_1120\n", "60854 Write about a character who’s stuck in an elev...\n", "60855 Write about a character who’s stuck in an elev...\n", "60856 Write about a character who’s stuck in an elev...\n", "60857 Write about a character who’s stuck in an elev...\n", "60858 Write about a character who’s stuck in an elev...\n", " ... \n", "61103 Write about a character who’s stuck in an elev...\n", "61104 Write about a character who’s stuck in an elev...\n", "61105 Write about a character who’s stuck in an elev...\n", "61106 Write about a character who’s stuck in an elev...\n", "61107 Write about a character who’s stuck in an elev...\n", "Name: prompt, Length: 254, dtype: object\n", "prompt_1089\n", "56188 Write about a family attempting to hide their ...\n", "56189 Write about a family attempting to hide their ...\n", "56190 Write about a family attempting to hide their ...\n", "56191 Write about a family attempting to hide their ...\n", "56192 Write about a family attempting to hide their ...\n", " ... \n", "56305 Write about a family attempting to hide their ...\n", "56306 Write about a family attempting to hide their ...\n", "56307 Write about a family attempting to hide their ...\n", "56308 Write about a family attempting to hide their ...\n", "56309 Write about a family attempting to hide their ...\n", "Name: prompt, Length: 122, dtype: object\n", "prompt_1087\n", "55709 Start your story with two characters deciding ...\n", "55710 Start your story with two characters deciding ...\n", "55711 Start your story with two characters deciding ...\n", "55712 Start your story with two characters deciding ...\n", "55713 Start your story with two characters deciding ...\n", " ... \n", "56027 Start your story with two characters deciding ...\n", "56028 Start your story with two characters deciding ...\n", "56029 Start your story with two characters deciding ...\n", "56030 Start your story with two characters deciding ...\n", "56031 Start your story with two characters deciding ...\n", "Name: prompt, Length: 323, dtype: object\n", "prompt_1111\n", "59385 Write a post-apocalyptic story that features z...\n", "59386 Write a post-apocalyptic story that features z...\n", "59387 Write a post-apocalyptic story that features z...\n", "59388 Write a post-apocalyptic story that features z...\n", "59389 Write a post-apocalyptic story that features z...\n", " ... \n", "59559 Write a post-apocalyptic story that features z...\n", "59560 Write a post-apocalyptic story that features z...\n", "59561 Write a post-apocalyptic story that features z...\n", "59562 Write a post-apocalyptic story that features z...\n", "59563 Write a post-apocalyptic story that features z...\n", "Name: prompt, Length: 179, dtype: object\n", "prompt_0937\n", "34660 Start your story with someone being presented ...\n", "34661 Start your story with someone being presented ...\n", "34662 Start your story with someone being presented ...\n", "34663 Start your story with someone being presented ...\n", "34664 Start your story with someone being presented ...\n", " ... \n", "34876 Start your story with someone being presented ...\n", "34877 Start your story with someone being presented ...\n", "34878 Start your story with someone being presented ...\n", "34879 Start your story with someone being presented ...\n", "34880 Start your story with someone being presented ...\n", "Name: prompt, Length: 221, dtype: object\n", "prompt_1024\n", "46870 Write a story set in the summer, when suddenly...\n", "46871 Write a story set in the summer, when suddenly...\n", "46872 Write a story set in the summer, when suddenly...\n", "46873 Write a story set in the summer, when suddenly...\n", "46874 Write a story set in the summer, when suddenly...\n", " ... \n", "47029 Write a story set in the summer, when suddenly...\n", "47030 Write a story set in the summer, when suddenly...\n", "47031 Write a story set in the summer, when suddenly...\n", "47032 Write a story set in the summer, when suddenly...\n", "47033 Write a story set in the summer, when suddenly...\n", "Name: prompt, Length: 164, dtype: object\n", "prompt_1091\n", "56566 Write a ghost story where there’s more going o...\n", "56567 Write a ghost story where there’s more going o...\n", "56568 Write a ghost story where there’s more going o...\n", "56569 Write a ghost story where there’s more going o...\n", "56570 Write a ghost story where there’s more going o...\n", " ... \n", "56789 Write a ghost story where there’s more going o...\n", "56790 Write a ghost story where there’s more going o...\n", "56791 Write a ghost story where there’s more going o...\n", "56792 Write a ghost story where there’s more going o...\n", "56793 Write a ghost story where there’s more going o...\n", "Name: prompt, Length: 228, dtype: object\n", "prompt_0968\n", "39373 Start the story with the absence of a sensory ...\n", "39374 Start the story with the absence of a sensory ...\n", "39375 Start the story with the absence of a sensory ...\n", "39376 Start the story with the absence of a sensory ...\n", "39377 Start the story with the absence of a sensory ...\n", "39378 Start the story with the absence of a sensory ...\n", "39379 Start the story with the absence of a sensory ...\n", "39380 Start the story with the absence of a sensory ...\n", "39381 Start the story with the absence of a sensory ...\n", "39382 Start the story with the absence of a sensory ...\n", "39383 Start the story with the absence of a sensory ...\n", "39384 Start the story with the absence of a sensory ...\n", "39385 Start the story with the absence of a sensory ...\n", "39386 Start the story with the absence of a sensory ...\n", "39387 Start the story with the absence of a sensory ...\n", "39388 Start the story with the absence of a sensory ...\n", "39389 Start the story with the absence of a sensory ...\n", "39390 Start the story with the absence of a sensory ...\n", "39391 Start the story with the absence of a sensory ...\n", "39392 Start the story with the absence of a sensory ...\n", "39393 Start the story with the absence of a sensory ...\n", "39394 Start the story with the absence of a sensory ...\n", "39395 Start the story with the absence of a sensory ...\n", "39396 Start the story with the absence of a sensory ...\n", "39397 Start the story with the absence of a sensory ...\n", "39398 Start the story with the absence of a sensory ...\n", "39399 Start the story with the absence of a sensory ...\n", "39400 Start the story with the absence of a sensory ...\n", "39401 Start the story with the absence of a sensory ...\n", "39402 Start the story with the absence of a sensory ...\n", "39403 Start the story with the absence of a sensory ...\n", "39404 Start the story with the absence of a sensory ...\n", "39405 Start the story with the absence of a sensory ...\n", "39406 Start the story with the absence of a sensory ...\n", "39407 Start the story with the absence of a sensory ...\n", "39408 Start the story with the absence of a sensory ...\n", "39409 Start the story with the absence of a sensory ...\n", "39410 Start the story with the absence of a sensory ...\n", "39411 Start the story with the absence of a sensory ...\n", "39412 Start the story with the absence of a sensory ...\n", "39413 Start the story with the absence of a sensory ...\n", "39414 Start the story with the absence of a sensory ...\n", "39415 Start the story with the absence of a sensory ...\n", "39416 Start the story with the absence of a sensory ...\n", "39417 Start the story with the absence of a sensory ...\n", "39418 Start the story with the absence of a sensory ...\n", "39419 Start the story with the absence of a sensory ...\n", "39420 Start the story with the absence of a sensory ...\n", "39421 Start the story with the absence of a sensory ...\n", "39422 Start the story with the absence of a sensory ...\n", "39423 Start the story with the absence of a sensory ...\n", "39424 Start the story with the absence of a sensory ...\n", "39425 Start the story with the absence of a sensory ...\n", "Name: prompt, dtype: object\n", "prompt_0117\n", "6151 Write about someone stuck in an endless cycle ...\n", "6152 Write about someone stuck in an endless cycle ...\n", "6153 Write about someone stuck in an endless cycle ...\n", "6154 Write about someone stuck in an endless cycle ...\n", "6155 Write about someone stuck in an endless cycle ...\n", " ... \n", "6217 Write about someone stuck in an endless cycle ...\n", "6218 Write about someone stuck in an endless cycle ...\n", "6219 Write about someone stuck in an endless cycle ...\n", "6220 Write about someone stuck in an endless cycle ...\n", "6221 Write about someone stuck in an endless cycle ...\n", "Name: prompt, Length: 71, dtype: object\n", "prompt_1142\n", "64471 Write a story about high school sweethearts co...\n", "64472 Write a story about high school sweethearts co...\n", "64473 Write a story about high school sweethearts co...\n", "64474 Write a story about high school sweethearts co...\n", "64475 Write a story about high school sweethearts co...\n", " ... \n", "64753 Write a story about high school sweethearts co...\n", "64754 Write a story about high school sweethearts co...\n", "64755 Write a story about high school sweethearts co...\n", "64756 Write a story about high school sweethearts co...\n", "64757 Write a story about high school sweethearts co...\n", "Name: prompt, Length: 287, dtype: object\n", "prompt_0146\n", "7765 Write a story entirely of dialogue. Nothing bu...\n", "7766 Write a story entirely of dialogue. Nothing bu...\n", "7767 Write a story entirely of dialogue. Nothing bu...\n", "7768 Write a story entirely of dialogue. Nothing bu...\n", "7769 Write a story entirely of dialogue. Nothing bu...\n", " ... \n", "8044 Write a story entirely of dialogue. Nothing bu...\n", "8045 Write a story entirely of dialogue. Nothing bu...\n", "8046 Write a story entirely of dialogue. Nothing bu...\n", "8047 Write a story entirely of dialogue. Nothing bu...\n", "8048 Write a story entirely of dialogue. Nothing bu...\n", "Name: prompt, Length: 284, dtype: object\n", "prompt_1005\n", "43705 Write about a first date that surprises both p...\n", "43706 Write about a first date that surprises both p...\n", "43707 Write about a first date that surprises both p...\n", "43708 Write about a first date that surprises both p...\n", "43709 Write about a first date that surprises both p...\n", " ... \n", "43862 Write about a first date that surprises both p...\n", "43863 Write about a first date that surprises both p...\n", "43864 Write about a first date that surprises both p...\n", "43865 Write about a first date that surprises both p...\n", "43866 Write about a first date that surprises both p...\n", "Name: prompt, Length: 162, dtype: object\n", "prompt_1084\n", "55202 Write about a vampire or werewolf who moves in...\n", "55203 Write about a vampire or werewolf who moves in...\n", "55204 Write about a vampire or werewolf who moves in...\n", "55205 Write about a vampire or werewolf who moves in...\n", "55206 Write about a vampire or werewolf who moves in...\n", " ... \n", "55359 Write about a vampire or werewolf who moves in...\n", "55360 Write about a vampire or werewolf who moves in...\n", "55361 Write about a vampire or werewolf who moves in...\n", "55362 Write about a vampire or werewolf who moves in...\n", "55363 Write about a vampire or werewolf who moves in...\n", "Name: prompt, Length: 162, dtype: object\n", "prompt_1172\n", "69960 Write about someone who has a superpower.\n", "69961 Write about someone who has a superpower.\n", "69962 Write about someone who has a superpower.\n", "69963 Write about someone who has a superpower.\n", "69964 Write about someone who has a superpower.\n", " ... \n", "70292 Write about someone who has a superpower.\n", "70293 Write about someone who has a superpower.\n", "70294 Write about someone who has a superpower.\n", "70295 Write about someone who has a superpower.\n", "70296 Write about someone who has a superpower.\n", "Name: prompt, Length: 337, dtype: object\n", "prompt_0944\n", "35521 Write your story about two characters tidying ...\n", "35522 Write your story about two characters tidying ...\n", "35523 Write your story about two characters tidying ...\n", "35524 Write your story about two characters tidying ...\n", "35525 Write your story about two characters tidying ...\n", " ... \n", "35668 Write your story about two characters tidying ...\n", "35669 Write your story about two characters tidying ...\n", "35670 Write your story about two characters tidying ...\n", "35671 Write your story about two characters tidying ...\n", "35672 Write your story about two characters tidying ...\n", "Name: prompt, Length: 152, dtype: object\n", "prompt_1005\n", "43705 Write about a first date that surprises both p...\n", "43706 Write about a first date that surprises both p...\n", "43707 Write about a first date that surprises both p...\n", "43708 Write about a first date that surprises both p...\n", "43709 Write about a first date that surprises both p...\n", " ... \n", "43862 Write about a first date that surprises both p...\n", "43863 Write about a first date that surprises both p...\n", "43864 Write about a first date that surprises both p...\n", "43865 Write about a first date that surprises both p...\n", "43866 Write about a first date that surprises both p...\n", "Name: prompt, Length: 162, dtype: object\n", "prompt_1027\n", "47435 Write about two people going sledding for the ...\n", "47436 Write about two people going sledding for the ...\n", "47437 Write about two people going sledding for the ...\n", "47438 Write about two people going sledding for the ...\n", "47439 Write about two people going sledding for the ...\n", " ... \n", "47561 Write about two people going sledding for the ...\n", "47562 Write about two people going sledding for the ...\n", "47563 Write about two people going sledding for the ...\n", "47564 Write about two people going sledding for the ...\n", "47565 Write about two people going sledding for the ...\n", "Name: prompt, Length: 131, dtype: object\n", "prompt_0985\n", "40711 Start your story with the line, “That’s the th...\n", "40712 Start your story with the line, “That’s the th...\n", "40713 Start your story with the line, “That’s the th...\n", "40714 Start your story with the line, “That’s the th...\n", "40715 Start your story with the line, “That’s the th...\n", " ... \n", "40992 Start your story with the line, “That’s the th...\n", "40993 Start your story with the line, “That’s the th...\n", "40994 Start your story with the line, “That’s the th...\n", "40995 Start your story with the line, “That’s the th...\n", "40996 Start your story with the line, “That’s the th...\n", "Name: prompt, Length: 286, dtype: object\n", "prompt_0985\n", "40711 Start your story with the line, “That’s the th...\n", "40712 Start your story with the line, “That’s the th...\n", "40713 Start your story with the line, “That’s the th...\n", "40714 Start your story with the line, “That’s the th...\n", "40715 Start your story with the line, “That’s the th...\n", " ... \n", "40992 Start your story with the line, “That’s the th...\n", "40993 Start your story with the line, “That’s the th...\n", "40994 Start your story with the line, “That’s the th...\n", "40995 Start your story with the line, “That’s the th...\n", "40996 Start your story with the line, “That’s the th...\n", "Name: prompt, Length: 286, dtype: object\n", "prompt_1069\n", "52899 Start your story with the line, “This was supp...\n", "52900 Start your story with the line, “This was supp...\n", "52901 Start your story with the line, “This was supp...\n", "52902 Start your story with the line, “This was supp...\n", "52903 Start your story with the line, “This was supp...\n", " ... \n", "53026 Start your story with the line, “This was supp...\n", "53027 Start your story with the line, “This was supp...\n", "53028 Start your story with the line, “This was supp...\n", "53029 Start your story with the line, “This was supp...\n", "53030 Start your story with the line, “This was supp...\n", "Name: prompt, Length: 132, dtype: object\n", "prompt_1150\n", "65712 You thought he was dead, but there he is, righ...\n", "65713 You thought he was dead, but there he is, righ...\n", "65714 You thought he was dead, but there he is, righ...\n", "65715 You thought he was dead, but there he is, righ...\n", "65716 You thought he was dead, but there he is, righ...\n", " ... \n", "66127 You thought he was dead, but there he is, righ...\n", "66128 You thought he was dead, but there he is, righ...\n", "66129 You thought he was dead, but there he is, righ...\n", "66130 You thought he was dead, but there he is, righ...\n", "66131 You thought he was dead, but there he is, righ...\n", "Name: prompt, Length: 420, dtype: object\n", "prompt_0953\n", "37637 Write about a character stumbling upon a libra...\n", "37638 Write about a character stumbling upon a libra...\n", "37639 Write about a character stumbling upon a libra...\n", "37640 Write about a character stumbling upon a libra...\n", "37641 Write about a character stumbling upon a libra...\n", " ... \n", "37842 Write about a character stumbling upon a libra...\n", "37843 Write about a character stumbling upon a libra...\n", "37844 Write about a character stumbling upon a libra...\n", "37845 Write about a character stumbling upon a libra...\n", "37846 Write about a character stumbling upon a libra...\n", "Name: prompt, Length: 210, dtype: object\n", "prompt_0919\n", "31610 Set your story on (or in) a winding river.\n", "31611 Set your story on (or in) a winding river.\n", "31612 Set your story on (or in) a winding river.\n", "31613 Set your story on (or in) a winding river.\n", "31614 Set your story on (or in) a winding river.\n", " ... \n", "31730 Set your story on (or in) a winding river.\n", "31731 Set your story on (or in) a winding river.\n", "31732 Set your story on (or in) a winding river.\n", "31733 Set your story on (or in) a winding river.\n", "31734 Set your story on (or in) a winding river.\n", "Name: prompt, Length: 125, dtype: object\n", "prompt_0952\n", "36994 Write a story that begins in the light and end...\n", "36995 Write a story that begins in the light and end...\n", "36996 Write a story that begins in the light and end...\n", "36997 Write a story that begins in the light and end...\n", "36998 Write a story that begins in the light and end...\n", " ... \n", "37376 Write a story that begins in the light and end...\n", "37377 Write a story that begins in the light and end...\n", "37378 Write a story that begins in the light and end...\n", "37379 Write a story that begins in the light and end...\n", "37380 Write a story that begins in the light and end...\n", "Name: prompt, Length: 387, dtype: object\n", "prompt_1160\n", "68088 Write a story about a person experiencing pre-...\n", "68089 Write a story about a person experiencing pre-...\n", "68090 Write a story about a person experiencing pre-...\n", "68091 Write a story about a person experiencing pre-...\n", "68092 Write a story about a person experiencing pre-...\n", " ... \n", "68245 Write a story about a person experiencing pre-...\n", "68246 Write a story about a person experiencing pre-...\n", "68247 Write a story about a person experiencing pre-...\n", "68248 Write a story about a person experiencing pre-...\n", "68249 Write a story about a person experiencing pre-...\n", "Name: prompt, Length: 162, dtype: object\n", "prompt_1176\n", "71095 \"Just say it,\" you silently reminded yourself....\n", "71096 \"Just say it,\" you silently reminded yourself....\n", "71097 \"Just say it,\" you silently reminded yourself....\n", "71098 \"Just say it,\" you silently reminded yourself....\n", "71099 \"Just say it,\" you silently reminded yourself....\n", " ... \n", "71311 \"Just say it,\" you silently reminded yourself....\n", "71312 \"Just say it,\" you silently reminded yourself....\n", "71313 \"Just say it,\" you silently reminded yourself....\n", "71314 \"Just say it,\" you silently reminded yourself....\n", "71315 \"Just say it,\" you silently reminded yourself....\n", "Name: prompt, Length: 221, dtype: object\n", "prompt_1116\n", "60115 Write about a character arriving in a place un...\n", "60116 Write about a character arriving in a place un...\n", "60117 Write about a character arriving in a place un...\n", "60118 Write about a character arriving in a place un...\n", "60119 Write about a character arriving in a place un...\n", " ... \n", "60311 Write about a character arriving in a place un...\n", "60312 Write about a character arriving in a place un...\n", "60313 Write about a character arriving in a place un...\n", "60314 Write about a character arriving in a place un...\n", "60315 Write about a character arriving in a place un...\n", "Name: prompt, Length: 201, dtype: object\n", "prompt_1115\n", "60046 Set your story in a place with extreme weather...\n", "60047 Set your story in a place with extreme weather...\n", "60048 Set your story in a place with extreme weather...\n", "60049 Set your story in a place with extreme weather...\n", "60050 Set your story in a place with extreme weather...\n", " ... \n", "60110 Set your story in a place with extreme weather...\n", "60111 Set your story in a place with extreme weather...\n", "60112 Set your story in a place with extreme weather...\n", "60113 Set your story in a place with extreme weather...\n", "60114 Set your story in a place with extreme weather...\n", "Name: prompt, Length: 69, dtype: object\n", "prompt_1165\n", "68891 Write a story about waiting — but don't reveal...\n", "68892 Write a story about waiting — but don't reveal...\n", "68893 Write a story about waiting — but don't reveal...\n", "68894 Write a story about waiting — but don't reveal...\n", "68895 Write a story about waiting — but don't reveal...\n", " ... \n", "69211 Write a story about waiting — but don't reveal...\n", "69212 Write a story about waiting — but don't reveal...\n", "69213 Write a story about waiting — but don't reveal...\n", "69214 Write a story about waiting — but don't reveal...\n", "69215 Write a story about waiting — but don't reveal...\n", "Name: prompt, Length: 325, dtype: object\n", "prompt_1077\n", "54378 Write about a pirate captain obsessed with fin...\n", "54379 Write about a pirate captain obsessed with fin...\n", "54380 Write about a pirate captain obsessed with fin...\n", "54381 Write about a pirate captain obsessed with fin...\n", "54382 Write about a pirate captain obsessed with fin...\n", " ... \n", "54518 Write about a pirate captain obsessed with fin...\n", "54519 Write about a pirate captain obsessed with fin...\n", "54520 Write about a pirate captain obsessed with fin...\n", "54521 Write about a pirate captain obsessed with fin...\n", "54522 Write about a pirate captain obsessed with fin...\n", "Name: prompt, Length: 145, dtype: object\n", "prompt_0931\n", "33661 Start your story with the arrival of a strange...\n", "33662 Start your story with the arrival of a strange...\n", "33663 Start your story with the arrival of a strange...\n", "33664 Start your story with the arrival of a strange...\n", "33665 Start your story with the arrival of a strange...\n", " ... \n", "33846 Start your story with the arrival of a strange...\n", "33847 Start your story with the arrival of a strange...\n", "33848 Start your story with the arrival of a strange...\n", "33849 Start your story with the arrival of a strange...\n", "33850 Start your story with the arrival of a strange...\n", "Name: prompt, Length: 190, dtype: object\n", "prompt_0960\n", "38500 Set your story in a world living with the cons...\n", "38501 Set your story in a world living with the cons...\n", "38502 Set your story in a world living with the cons...\n", "38503 Set your story in a world living with the cons...\n", "38504 Set your story in a world living with the cons...\n", " ... \n", "38630 Set your story in a world living with the cons...\n", "38631 Set your story in a world living with the cons...\n", "38632 Set your story in a world living with the cons...\n", "38633 Set your story in a world living with the cons...\n", "38634 Set your story in a world living with the cons...\n", "Name: prompt, Length: 135, dtype: object\n", "prompt_1189\n", "72488 Write a story that starts with a character-rev...\n", "72489 Write a story that starts with a character-rev...\n", "72490 Write a story that starts with a character-rev...\n", "72491 Write a story that starts with a character-rev...\n", "72492 Write a story that starts with a character-rev...\n", " ... \n", "72626 Write a story that starts with a character-rev...\n", "72627 Write a story that starts with a character-rev...\n", "72628 Write a story that starts with a character-rev...\n", "72629 Write a story that starts with a character-rev...\n", "72630 Write a story that starts with a character-rev...\n", "Name: prompt, Length: 143, dtype: object\n", "prompt_0146\n", "7765 Write a story entirely of dialogue. Nothing bu...\n", "7766 Write a story entirely of dialogue. Nothing bu...\n", "7767 Write a story entirely of dialogue. Nothing bu...\n", "7768 Write a story entirely of dialogue. Nothing bu...\n", "7769 Write a story entirely of dialogue. Nothing bu...\n", " ... \n", "8044 Write a story entirely of dialogue. Nothing bu...\n", "8045 Write a story entirely of dialogue. Nothing bu...\n", "8046 Write a story entirely of dialogue. Nothing bu...\n", "8047 Write a story entirely of dialogue. Nothing bu...\n", "8048 Write a story entirely of dialogue. Nothing bu...\n", "Name: prompt, Length: 284, dtype: object\n", "prompt_0932\n", "33393 Write a story about strangers becoming friends...\n", "33394 Write a story about strangers becoming friends...\n", "33395 Write a story about strangers becoming friends...\n", "33396 Write a story about strangers becoming friends...\n", "33397 Write a story about strangers becoming friends...\n", " ... \n", "33656 Write a story about strangers becoming friends...\n", "33657 Write a story about strangers becoming friends...\n", "33658 Write a story about strangers becoming friends...\n", "33659 Write a story about strangers becoming friends...\n", "33660 Write a story about strangers becoming friends...\n", "Name: prompt, Length: 268, dtype: object\n", "prompt_1142\n", "64471 Write a story about high school sweethearts co...\n", "64472 Write a story about high school sweethearts co...\n", "64473 Write a story about high school sweethearts co...\n", "64474 Write a story about high school sweethearts co...\n", "64475 Write a story about high school sweethearts co...\n", " ... \n", "64753 Write a story about high school sweethearts co...\n", "64754 Write a story about high school sweethearts co...\n", "64755 Write a story about high school sweethearts co...\n", "64756 Write a story about high school sweethearts co...\n", "64757 Write a story about high school sweethearts co...\n", "Name: prompt, Length: 287, dtype: object\n", "prompt_1096\n", "57385 Start your story with the line, “By the time I...\n", "57386 Start your story with the line, “By the time I...\n", "57387 Start your story with the line, “By the time I...\n", "57388 Start your story with the line, “By the time I...\n", "57389 Start your story with the line, “By the time I...\n", " ... \n", "57706 Start your story with the line, “By the time I...\n", "57707 Start your story with the line, “By the time I...\n", "57708 Start your story with the line, “By the time I...\n", "57709 Start your story with the line, “By the time I...\n", "57710 Start your story with the line, “By the time I...\n", "Name: prompt, Length: 326, dtype: object\n", "prompt_1177\n", "70881 As you check your mail, you notice a letter th...\n", "70882 As you check your mail, you notice a letter th...\n", "70883 As you check your mail, you notice a letter th...\n", "70884 As you check your mail, you notice a letter th...\n", "70885 As you check your mail, you notice a letter th...\n", " ... \n", "71090 As you check your mail, you notice a letter th...\n", "71091 As you check your mail, you notice a letter th...\n", "71092 As you check your mail, you notice a letter th...\n", "71093 As you check your mail, you notice a letter th...\n", "71094 As you check your mail, you notice a letter th...\n", "Name: prompt, Length: 214, dtype: object\n", "prompt_1168\n", "69648 Write a story that features a protagonist with...\n", "69649 Write a story that features a protagonist with...\n", "69650 Write a story that features a protagonist with...\n", "69651 Write a story that features a protagonist with...\n", "69652 Write a story that features a protagonist with...\n", " ... \n", "69756 Write a story that features a protagonist with...\n", "69757 Write a story that features a protagonist with...\n", "69758 Write a story that features a protagonist with...\n", "69759 Write a story that features a protagonist with...\n", "69760 Write a story that features a protagonist with...\n", "Name: prompt, Length: 113, dtype: object\n", "prompt_1158\n", "67634 Write a story about a summer afternoon spent i...\n", "67635 Write a story about a summer afternoon spent i...\n", "67636 Write a story about a summer afternoon spent i...\n", "67637 Write a story about a summer afternoon spent i...\n", "67638 Write a story about a summer afternoon spent i...\n", " ... \n", "67937 Write a story about a summer afternoon spent i...\n", "67938 Write a story about a summer afternoon spent i...\n", "67939 Write a story about a summer afternoon spent i...\n", "67940 Write a story about a summer afternoon spent i...\n", "67941 Write a story about a summer afternoon spent i...\n", "Name: prompt, Length: 308, dtype: object\n", "prompt_0937\n", "34660 Start your story with someone being presented ...\n", "34661 Start your story with someone being presented ...\n", "34662 Start your story with someone being presented ...\n", "34663 Start your story with someone being presented ...\n", "34664 Start your story with someone being presented ...\n", " ... \n", "34876 Start your story with someone being presented ...\n", "34877 Start your story with someone being presented ...\n", "34878 Start your story with someone being presented ...\n", "34879 Start your story with someone being presented ...\n", "34880 Start your story with someone being presented ...\n", "Name: prompt, Length: 221, dtype: object\n", "prompt_0044\n", "2338 Write about a person who constantly has to put...\n", "2339 Write about a person who constantly has to put...\n", "2340 Write about a person who constantly has to put...\n", "2341 Write about a person who constantly has to put...\n", "2342 Write about a person who constantly has to put...\n", " ... \n", "2416 Write about a person who constantly has to put...\n", "2417 Write about a person who constantly has to put...\n", "2418 Write about a person who constantly has to put...\n", "2419 Write about a person who constantly has to put...\n", "2420 Write about a person who constantly has to put...\n", "Name: prompt, Length: 83, dtype: object\n", "prompt_1142\n", "64471 Write a story about high school sweethearts co...\n", "64472 Write a story about high school sweethearts co...\n", "64473 Write a story about high school sweethearts co...\n", "64474 Write a story about high school sweethearts co...\n", "64475 Write a story about high school sweethearts co...\n", " ... \n", "64753 Write a story about high school sweethearts co...\n", "64754 Write a story about high school sweethearts co...\n", "64755 Write a story about high school sweethearts co...\n", "64756 Write a story about high school sweethearts co...\n", "64757 Write a story about high school sweethearts co...\n", "Name: prompt, Length: 287, dtype: object\n", "prompt_1167\n", "69390 Write a story that takes place in a waiting room.\n", "69391 Write a story that takes place in a waiting room.\n", "69392 Write a story that takes place in a waiting room.\n", "69393 Write a story that takes place in a waiting room.\n", "69394 Write a story that takes place in a waiting room.\n", " ... \n", "69643 Write a story that takes place in a waiting room.\n", "69644 Write a story that takes place in a waiting room.\n", "69645 Write a story that takes place in a waiting room.\n", "69646 Write a story that takes place in a waiting room.\n", "69647 Write a story that takes place in a waiting room.\n", "Name: prompt, Length: 258, dtype: object\n", "prompt_1274\n", "81742 Write a short story that ends with a twist.\n", "81743 Write a short story that ends with a twist.\n", "81744 Write a short story that ends with a twist.\n", "81745 Write a short story that ends with a twist.\n", "81746 Write a short story that ends with a twist.\n", " ... \n", "81894 Write a short story that ends with a twist.\n", "81895 Write a short story that ends with a twist.\n", "81896 Write a short story that ends with a twist.\n", "81897 Write a short story that ends with a twist.\n", "81898 Write a short story that ends with a twist.\n", "Name: prompt, Length: 157, dtype: object\n", "prompt_1017\n", "45613 Write about someone who decides it’s time to c...\n", "45614 Write about someone who decides it’s time to c...\n", "45615 Write about someone who decides it’s time to c...\n", "45616 Write about someone who decides it’s time to c...\n", "45617 Write about someone who decides it’s time to c...\n", " ... \n", "45873 Write about someone who decides it’s time to c...\n", "45874 Write about someone who decides it’s time to c...\n", "45875 Write about someone who decides it’s time to c...\n", "45876 Write about someone who decides it’s time to c...\n", "45877 Write about someone who decides it’s time to c...\n", "Name: prompt, Length: 265, dtype: object\n", "prompt_0638\n", "15353 Write a story featuring an element of time-tra...\n", "15354 Write a story featuring an element of time-tra...\n", "15355 Write a story featuring an element of time-tra...\n", "15356 Write a story featuring an element of time-tra...\n", "15357 Write a story featuring an element of time-tra...\n", " ... \n", "15416 Write a story featuring an element of time-tra...\n", "15417 Write a story featuring an element of time-tra...\n", "15418 Write a story featuring an element of time-tra...\n", "15419 Write a story featuring an element of time-tra...\n", "15420 Write a story featuring an element of time-tra...\n", "Name: prompt, Length: 68, dtype: object\n", "prompt_1197\n", "73639 Write a story about transformation.\n", "73640 Write a story about transformation.\n", "73641 Write a story about transformation.\n", "73642 Write a story about transformation.\n", "73643 Write a story about transformation.\n", " ... \n", "73807 Write a story about transformation.\n", "73808 Write a story about transformation.\n", "73809 Write a story about transformation.\n", "73810 Write a story about transformation.\n", "73811 Write a story about transformation.\n", "Name: prompt, Length: 173, dtype: object\n", "prompt_0935\n", "34213 A character stands in front of two doors. Writ...\n", "34214 A character stands in front of two doors. Writ...\n", "34215 A character stands in front of two doors. Writ...\n", "34216 A character stands in front of two doors. Writ...\n", "34217 A character stands in front of two doors. Writ...\n", " ... \n", "34448 A character stands in front of two doors. Writ...\n", "34449 A character stands in front of two doors. Writ...\n", "34450 A character stands in front of two doors. Writ...\n", "34451 A character stands in front of two doors. Writ...\n", "34452 A character stands in front of two doors. Writ...\n", "Name: prompt, Length: 240, dtype: object\n", "prompt_0620\n", "14572 Write a story where the law plays an important...\n", "14573 Write a story where the law plays an important...\n", "14574 Write a story where the law plays an important...\n", "14575 Write a story where the law plays an important...\n", "14576 Write a story where the law plays an important...\n", "14577 Write a story where the law plays an important...\n", "14578 Write a story where the law plays an important...\n", "14579 Write a story where the law plays an important...\n", "14580 Write a story where the law plays an important...\n", "14581 Write a story where the law plays an important...\n", "14582 Write a story where the law plays an important...\n", "14583 Write a story where the law plays an important...\n", "14584 Write a story where the law plays an important...\n", "14585 Write a story where the law plays an important...\n", "14586 Write a story where the law plays an important...\n", "14587 Write a story where the law plays an important...\n", "14588 Write a story where the law plays an important...\n", "14589 Write a story where the law plays an important...\n", "14590 Write a story where the law plays an important...\n", "14591 Write a story where the law plays an important...\n", "14592 Write a story where the law plays an important...\n", "14593 Write a story where the law plays an important...\n", "14594 Write a story where the law plays an important...\n", "14595 Write a story where the law plays an important...\n", "Name: prompt, dtype: object\n", "prompt_0923\n", "32047 Set your story within a window of opportunity,...\n", "32048 Set your story within a window of opportunity,...\n", "32049 Set your story within a window of opportunity,...\n", "32050 Set your story within a window of opportunity,...\n", "32051 Set your story within a window of opportunity,...\n", " ... \n", "32189 Set your story within a window of opportunity,...\n", "32190 Set your story within a window of opportunity,...\n", "32191 Set your story within a window of opportunity,...\n", "32192 Set your story within a window of opportunity,...\n", "32193 Set your story within a window of opportunity,...\n", "Name: prompt, Length: 147, dtype: object\n", "prompt_0912\n", "29891 Start or end your story with two characters si...\n", "29892 Start or end your story with two characters si...\n", "29893 Start or end your story with two characters si...\n", "29894 Start or end your story with two characters si...\n", "29895 Start or end your story with two characters si...\n", " ... \n", "30153 Start or end your story with two characters si...\n", "30154 Start or end your story with two characters si...\n", "30155 Start or end your story with two characters si...\n", "30156 Start or end your story with two characters si...\n", "30157 Start or end your story with two characters si...\n", "Name: prompt, Length: 267, dtype: object\n", "prompt_1032\n", "48007 Write a story told exclusively through dialogue.\n", "48008 Write a story told exclusively through dialogue.\n", "48009 Write a story told exclusively through dialogue.\n", "48010 Write a story told exclusively through dialogue.\n", "48011 Write a story told exclusively through dialogue.\n", " ... \n", "48356 Write a story told exclusively through dialogue.\n", "48357 Write a story told exclusively through dialogue.\n", "48358 Write a story told exclusively through dialogue.\n", "48359 Write a story told exclusively through dialogue.\n", "48360 Write a story told exclusively through dialogue.\n", "Name: prompt, Length: 354, dtype: object\n", "prompt_1225\n", "77452 Write a story that takes place in the woods.\n", "77453 Write a story that takes place in the woods.\n", "77454 Write a story that takes place in the woods.\n", "77455 Write a story that takes place in the woods.\n", "77456 Write a story that takes place in the woods.\n", " ... \n", "77628 Write a story that takes place in the woods.\n", "77629 Write a story that takes place in the woods.\n", "77630 Write a story that takes place in the woods.\n", "77631 Write a story that takes place in the woods.\n", "77632 Write a story that takes place in the woods.\n", "Name: prompt, Length: 181, dtype: object\n", "prompt_0958\n", "38022 Start your story with someone sitting on a cro...\n", "38023 Start your story with someone sitting on a cro...\n", "38024 Start your story with someone sitting on a cro...\n", "38025 Start your story with someone sitting on a cro...\n", "38026 Start your story with someone sitting on a cro...\n", " ... \n", "38279 Start your story with someone sitting on a cro...\n", "38280 Start your story with someone sitting on a cro...\n", "38281 Start your story with someone sitting on a cro...\n", "38282 Start your story with someone sitting on a cro...\n", "38283 Start your story with someone sitting on a cro...\n", "Name: prompt, Length: 262, dtype: object\n", "prompt_1013\n", "44965 Write about a “found family” who are finally a...\n", "44966 Write about a “found family” who are finally a...\n", "44967 Write about a “found family” who are finally a...\n", "44968 Write about a “found family” who are finally a...\n", "44969 Write about a “found family” who are finally a...\n", " ... \n", "45061 Write about a “found family” who are finally a...\n", "45062 Write about a “found family” who are finally a...\n", "45063 Write about a “found family” who are finally a...\n", "45064 Write about a “found family” who are finally a...\n", "45065 Write about a “found family” who are finally a...\n", "Name: prompt, Length: 101, dtype: object\n", "prompt_0998\n", "42817 Write about an android just trying to blend in...\n", "42818 Write about an android just trying to blend in...\n", "42819 Write about an android just trying to blend in...\n", "42820 Write about an android just trying to blend in...\n", "42821 Write about an android just trying to blend in...\n", " ... \n", "43037 Write about an android just trying to blend in...\n", "43038 Write about an android just trying to blend in...\n", "43039 Write about an android just trying to blend in...\n", "43040 Write about an android just trying to blend in...\n", "43041 Write about an android just trying to blend in...\n", "Name: prompt, Length: 225, dtype: object\n", "prompt_0986\n", "40997 Write a story that takes place in the same bui...\n", "40998 Write a story that takes place in the same bui...\n", "40999 Write a story that takes place in the same bui...\n", "41000 Write a story that takes place in the same bui...\n", "41001 Write a story that takes place in the same bui...\n", " ... \n", "41148 Write a story that takes place in the same bui...\n", "41149 Write a story that takes place in the same bui...\n", "41150 Write a story that takes place in the same bui...\n", "41151 Write a story that takes place in the same bui...\n", "41152 Write a story that takes place in the same bui...\n", "Name: prompt, Length: 156, dtype: object\n", "prompt_0980\n", "40220 Write a fairy tale about someone who can commu...\n", "40221 Write a fairy tale about someone who can commu...\n", "40222 Write a fairy tale about someone who can commu...\n", "40223 Write a fairy tale about someone who can commu...\n", "40224 Write a fairy tale about someone who can commu...\n", " ... \n", "40384 Write a fairy tale about someone who can commu...\n", "40385 Write a fairy tale about someone who can commu...\n", "40386 Write a fairy tale about someone who can commu...\n", "40387 Write a fairy tale about someone who can commu...\n", "40388 Write a fairy tale about someone who can commu...\n", "Name: prompt, Length: 169, dtype: object\n", "prompt_1168\n", "69648 Write a story that features a protagonist with...\n", "69649 Write a story that features a protagonist with...\n", "69650 Write a story that features a protagonist with...\n", "69651 Write a story that features a protagonist with...\n", "69652 Write a story that features a protagonist with...\n", " ... \n", "69756 Write a story that features a protagonist with...\n", "69757 Write a story that features a protagonist with...\n", "69758 Write a story that features a protagonist with...\n", "69759 Write a story that features a protagonist with...\n", "69760 Write a story that features a protagonist with...\n", "Name: prompt, Length: 113, dtype: object\n", "prompt_1105\n", "58423 Write about a character who smells something f...\n", "58424 Write about a character who smells something f...\n", "58425 Write about a character who smells something f...\n", "58426 Write about a character who smells something f...\n", "58427 Write about a character who smells something f...\n", " ... \n", "58639 Write about a character who smells something f...\n", "58640 Write about a character who smells something f...\n", "58641 Write about a character who smells something f...\n", "58642 Write about a character who smells something f...\n", "58643 Write about a character who smells something f...\n", "Name: prompt, Length: 221, dtype: object\n", "prompt_1159\n", "67942 Write a story told entirely through one chase ...\n", "67943 Write a story told entirely through one chase ...\n", "67944 Write a story told entirely through one chase ...\n", "67945 Write a story told entirely through one chase ...\n", "67946 Write a story told entirely through one chase ...\n", " ... \n", "68083 Write a story told entirely through one chase ...\n", "68084 Write a story told entirely through one chase ...\n", "68085 Write a story told entirely through one chase ...\n", "68086 Write a story told entirely through one chase ...\n", "68087 Write a story told entirely through one chase ...\n", "Name: prompt, Length: 146, dtype: object\n", "prompt_1186\n", "72443 Write a story about activism.\n", "72444 Write a story about activism.\n", "72445 Write a story about activism.\n", "72446 Write a story about activism.\n", "72447 Write a story about activism.\n", "72448 Write a story about activism.\n", "72449 Write a story about activism.\n", "72450 Write a story about activism.\n", "72451 Write a story about activism.\n", "72452 Write a story about activism.\n", "72453 Write a story about activism.\n", "72454 Write a story about activism.\n", "72455 Write a story about activism.\n", "72456 Write a story about activism.\n", "72457 Write a story about activism.\n", "72458 Write a story about activism.\n", "72459 Write a story about activism.\n", "72460 Write a story about activism.\n", "72461 Write a story about activism.\n", "72462 Write a story about activism.\n", "72463 Write a story about activism.\n", "72464 Write a story about activism.\n", "72465 Write a story about activism.\n", "72466 Write a story about activism.\n", "72467 Write a story about activism.\n", "72468 Write a story about activism.\n", "72469 Write a story about activism.\n", "72470 Write a story about activism.\n", "72471 Write a story about activism.\n", "72472 Write a story about activism.\n", "72473 Write a story about activism.\n", "72474 Write a story about activism.\n", "72475 Write a story about activism.\n", "72476 Write a story about activism.\n", "72477 Write a story about activism.\n", "72478 Write a story about activism.\n", "72479 Write a story about activism.\n", "72480 Write a story about activism.\n", "72481 Write a story about activism.\n", "72482 Write a story about activism.\n", "72483 Write a story about activism.\n", "72484 Write a story about activism.\n", "72485 Write a story about activism.\n", "72486 Write a story about activism.\n", "72487 Write a story about activism.\n", "Name: prompt, dtype: object\n", "prompt_1114\n", "59814 Write a story that feels lonely, despite being...\n", "59815 Write a story that feels lonely, despite being...\n", "59816 Write a story that feels lonely, despite being...\n", "59817 Write a story that feels lonely, despite being...\n", "59818 Write a story that feels lonely, despite being...\n", " ... \n", "60041 Write a story that feels lonely, despite being...\n", "60042 Write a story that feels lonely, despite being...\n", "60043 Write a story that feels lonely, despite being...\n", "60044 Write a story that feels lonely, despite being...\n", "60045 Write a story that feels lonely, despite being...\n", "Name: prompt, Length: 232, dtype: object\n", "prompt_0701\n", "18109 Write a story about a librarian that doesn’t f...\n", "18110 Write a story about a librarian that doesn’t f...\n", "18111 Write a story about a librarian that doesn’t f...\n", "18112 Write a story about a librarian that doesn’t f...\n", "18113 Write a story about a librarian that doesn’t f...\n", " ... \n", "18168 Write a story about a librarian that doesn’t f...\n", "18169 Write a story about a librarian that doesn’t f...\n", "18170 Write a story about a librarian that doesn’t f...\n", "18171 Write a story about a librarian that doesn’t f...\n", "18172 Write a story about a librarian that doesn’t f...\n", "Name: prompt, Length: 64, dtype: object\n", "prompt_1061\n", "52046 Write about a character who everyone thinks is...\n", "52047 Write about a character who everyone thinks is...\n", "52048 Write about a character who everyone thinks is...\n", "52049 Write about a character who everyone thinks is...\n", "52050 Write about a character who everyone thinks is...\n", " ... \n", "52222 Write about a character who everyone thinks is...\n", "52223 Write about a character who everyone thinks is...\n", "52224 Write about a character who everyone thinks is...\n", "52225 Write about a character who everyone thinks is...\n", "52226 Write about a character who everyone thinks is...\n", "Name: prompt, Length: 181, dtype: object\n", "prompt_0925\n", "32333 Write a story that involves a magic window — o...\n", "32334 Write a story that involves a magic window — o...\n", "32335 Write a story that involves a magic window — o...\n", "32336 Write a story that involves a magic window — o...\n", "32337 Write a story that involves a magic window — o...\n", " ... \n", "32484 Write a story that involves a magic window — o...\n", "32485 Write a story that involves a magic window — o...\n", "32486 Write a story that involves a magic window — o...\n", "32487 Write a story that involves a magic window — o...\n", "32488 Write a story that involves a magic window — o...\n", "Name: prompt, Length: 156, dtype: object\n", "prompt_1018\n", "45878 Write about someone who keeps picking up diffe...\n", "45879 Write about someone who keeps picking up diffe...\n", "45880 Write about someone who keeps picking up diffe...\n", "45881 Write about someone who keeps picking up diffe...\n", "45882 Write about someone who keeps picking up diffe...\n", " ... \n", "46050 Write about someone who keeps picking up diffe...\n", "46051 Write about someone who keeps picking up diffe...\n", "46052 Write about someone who keeps picking up diffe...\n", "46053 Write about someone who keeps picking up diffe...\n", "46054 Write about someone who keeps picking up diffe...\n", "Name: prompt, Length: 177, dtype: object\n", "prompt_1094\n", "57046 Write a story from the perspective of a bird m...\n", "57047 Write a story from the perspective of a bird m...\n", "57048 Write a story from the perspective of a bird m...\n", "57049 Write a story from the perspective of a bird m...\n", "57050 Write a story from the perspective of a bird m...\n", " ... \n", "57171 Write a story from the perspective of a bird m...\n", "57172 Write a story from the perspective of a bird m...\n", "57173 Write a story from the perspective of a bird m...\n", "57174 Write a story from the perspective of a bird m...\n", "57175 Write a story from the perspective of a bird m...\n", "Name: prompt, Length: 130, dtype: object\n", "prompt_1167\n", "69390 Write a story that takes place in a waiting room.\n", "69391 Write a story that takes place in a waiting room.\n", "69392 Write a story that takes place in a waiting room.\n", "69393 Write a story that takes place in a waiting room.\n", "69394 Write a story that takes place in a waiting room.\n", " ... \n", "69643 Write a story that takes place in a waiting room.\n", "69644 Write a story that takes place in a waiting room.\n", "69645 Write a story that takes place in a waiting room.\n", "69646 Write a story that takes place in a waiting room.\n", "69647 Write a story that takes place in a waiting room.\n", "Name: prompt, Length: 258, dtype: object\n", "prompt_1017\n", "45613 Write about someone who decides it’s time to c...\n", "45614 Write about someone who decides it’s time to c...\n", "45615 Write about someone who decides it’s time to c...\n", "45616 Write about someone who decides it’s time to c...\n", "45617 Write about someone who decides it’s time to c...\n", " ... \n", "45873 Write about someone who decides it’s time to c...\n", "45874 Write about someone who decides it’s time to c...\n", "45875 Write about someone who decides it’s time to c...\n", "45876 Write about someone who decides it’s time to c...\n", "45877 Write about someone who decides it’s time to c...\n", "Name: prompt, Length: 265, dtype: object\n", "prompt_1072\n", "53481 Start your story with two characters watching ...\n", "53482 Start your story with two characters watching ...\n", "53483 Start your story with two characters watching ...\n", "53484 Start your story with two characters watching ...\n", "53485 Start your story with two characters watching ...\n", " ... \n", "53780 Start your story with two characters watching ...\n", "53781 Start your story with two characters watching ...\n", "53782 Start your story with two characters watching ...\n", "53783 Start your story with two characters watching ...\n", "53784 Start your story with two characters watching ...\n", "Name: prompt, Length: 304, dtype: object\n", "prompt_0796\n", "22465 Write about a character who always wears a mas...\n", "22466 Write about a character who always wears a mas...\n", "22467 Write about a character who always wears a mas...\n", "22468 Write about a character who always wears a mas...\n", "22469 Write about a character who always wears a mas...\n", "22470 Write about a character who always wears a mas...\n", "22471 Write about a character who always wears a mas...\n", "22472 Write about a character who always wears a mas...\n", "22473 Write about a character who always wears a mas...\n", "22474 Write about a character who always wears a mas...\n", "22475 Write about a character who always wears a mas...\n", "22476 Write about a character who always wears a mas...\n", "22477 Write about a character who always wears a mas...\n", "22478 Write about a character who always wears a mas...\n", "22479 Write about a character who always wears a mas...\n", "22480 Write about a character who always wears a mas...\n", "22481 Write about a character who always wears a mas...\n", "22482 Write about a character who always wears a mas...\n", "22483 Write about a character who always wears a mas...\n", "22484 Write about a character who always wears a mas...\n", "22485 Write about a character who always wears a mas...\n", "22486 Write about a character who always wears a mas...\n", "22487 Write about a character who always wears a mas...\n", "22488 Write about a character who always wears a mas...\n", "22489 Write about a character who always wears a mas...\n", "22490 Write about a character who always wears a mas...\n", "22491 Write about a character who always wears a mas...\n", "22492 Write about a character who always wears a mas...\n", "22493 Write about a character who always wears a mas...\n", "22494 Write about a character who always wears a mas...\n", "22495 Write about a character who always wears a mas...\n", "22496 Write about a character who always wears a mas...\n", "22497 Write about a character who always wears a mas...\n", "22498 Write about a character who always wears a mas...\n", "22499 Write about a character who always wears a mas...\n", "22500 Write about a character who always wears a mas...\n", "22501 Write about a character who always wears a mas...\n", "22502 Write about a character who always wears a mas...\n", "22503 Write about a character who always wears a mas...\n", "22504 Write about a character who always wears a mas...\n", "22505 Write about a character who always wears a mas...\n", "22506 Write about a character who always wears a mas...\n", "22507 Write about a character who always wears a mas...\n", "22508 Write about a character who always wears a mas...\n", "22509 Write about a character who always wears a mas...\n", "22510 Write about a character who always wears a mas...\n", "22511 Write about a character who always wears a mas...\n", "22512 Write about a character who always wears a mas...\n", "22513 Write about a character who always wears a mas...\n", "Name: prompt, dtype: object\n", "prompt_0862\n", "25026 Set your story in a roadside diner.\n", "25027 Set your story in a roadside diner.\n", "25028 Set your story in a roadside diner.\n", "25029 Set your story in a roadside diner.\n", "25030 Set your story in a roadside diner.\n", "25031 Set your story in a roadside diner.\n", "25032 Set your story in a roadside diner.\n", "25033 Set your story in a roadside diner.\n", "25034 Set your story in a roadside diner.\n", "25035 Set your story in a roadside diner.\n", "25036 Set your story in a roadside diner.\n", "25037 Set your story in a roadside diner.\n", "25038 Set your story in a roadside diner.\n", "25039 Set your story in a roadside diner.\n", "25040 Set your story in a roadside diner.\n", "25041 Set your story in a roadside diner.\n", "25042 Set your story in a roadside diner.\n", "25043 Set your story in a roadside diner.\n", "25044 Set your story in a roadside diner.\n", "25045 Set your story in a roadside diner.\n", "25046 Set your story in a roadside diner.\n", "25047 Set your story in a roadside diner.\n", "25048 Set your story in a roadside diner.\n", "25049 Set your story in a roadside diner.\n", "25050 Set your story in a roadside diner.\n", "25051 Set your story in a roadside diner.\n", "25052 Set your story in a roadside diner.\n", "25053 Set your story in a roadside diner.\n", "25054 Set your story in a roadside diner.\n", "25055 Set your story in a roadside diner.\n", "25056 Set your story in a roadside diner.\n", "25057 Set your story in a roadside diner.\n", "25058 Set your story in a roadside diner.\n", "25059 Set your story in a roadside diner.\n", "25060 Set your story in a roadside diner.\n", "25061 Set your story in a roadside diner.\n", "25062 Set your story in a roadside diner.\n", "25063 Set your story in a roadside diner.\n", "25064 Set your story in a roadside diner.\n", "25065 Set your story in a roadside diner.\n", "25066 Set your story in a roadside diner.\n", "25067 Set your story in a roadside diner.\n", "25068 Set your story in a roadside diner.\n", "25069 Set your story in a roadside diner.\n", "25070 Set your story in a roadside diner.\n", "25071 Set your story in a roadside diner.\n", "25072 Set your story in a roadside diner.\n", "25073 Set your story in a roadside diner.\n", "25074 Set your story in a roadside diner.\n", "25075 Set your story in a roadside diner.\n", "25076 Set your story in a roadside diner.\n", "25077 Set your story in a roadside diner.\n", "25078 Set your story in a roadside diner.\n", "25079 Set your story in a roadside diner.\n", "25080 Set your story in a roadside diner.\n", "25081 Set your story in a roadside diner.\n", "25082 Set your story in a roadside diner.\n", "25083 Set your story in a roadside diner.\n", "25084 Set your story in a roadside diner.\n", "Name: prompt, dtype: object\n", "prompt_1010\n", "44580 Start your story with someone entering a museu...\n", "44581 Start your story with someone entering a museu...\n", "44582 Start your story with someone entering a museu...\n", "44583 Start your story with someone entering a museu...\n", "44584 Start your story with someone entering a museu...\n", "44585 Start your story with someone entering a museu...\n", "44586 Start your story with someone entering a museu...\n", "44587 Start your story with someone entering a museu...\n", "44588 Start your story with someone entering a museu...\n", "44589 Start your story with someone entering a museu...\n", "44590 Start your story with someone entering a museu...\n", "44591 Start your story with someone entering a museu...\n", "44592 Start your story with someone entering a museu...\n", "44593 Start your story with someone entering a museu...\n", "44594 Start your story with someone entering a museu...\n", "44595 Start your story with someone entering a museu...\n", "44596 Start your story with someone entering a museu...\n", "44597 Start your story with someone entering a museu...\n", "44598 Start your story with someone entering a museu...\n", "44599 Start your story with someone entering a museu...\n", "44600 Start your story with someone entering a museu...\n", "44601 Start your story with someone entering a museu...\n", "44602 Start your story with someone entering a museu...\n", "44603 Start your story with someone entering a museu...\n", "44604 Start your story with someone entering a museu...\n", "44605 Start your story with someone entering a museu...\n", "44606 Start your story with someone entering a museu...\n", "44607 Start your story with someone entering a museu...\n", "44608 Start your story with someone entering a museu...\n", "44609 Start your story with someone entering a museu...\n", "44610 Start your story with someone entering a museu...\n", "44611 Start your story with someone entering a museu...\n", "44612 Start your story with someone entering a museu...\n", "44613 Start your story with someone entering a museu...\n", "44614 Start your story with someone entering a museu...\n", "44615 Start your story with someone entering a museu...\n", "44616 Start your story with someone entering a museu...\n", "44617 Start your story with someone entering a museu...\n", "44618 Start your story with someone entering a museu...\n", "44619 Start your story with someone entering a museu...\n", "44620 Start your story with someone entering a museu...\n", "Name: prompt, dtype: object\n", "prompt_1069\n", "52899 Start your story with the line, “This was supp...\n", "52900 Start your story with the line, “This was supp...\n", "52901 Start your story with the line, “This was supp...\n", "52902 Start your story with the line, “This was supp...\n", "52903 Start your story with the line, “This was supp...\n", " ... \n", "53026 Start your story with the line, “This was supp...\n", "53027 Start your story with the line, “This was supp...\n", "53028 Start your story with the line, “This was supp...\n", "53029 Start your story with the line, “This was supp...\n", "53030 Start your story with the line, “This was supp...\n", "Name: prompt, Length: 132, dtype: object\n", "prompt_1155\n", "66626 Write a story that involves a mystery — it doe...\n", "66627 Write a story that involves a mystery — it doe...\n", "66628 Write a story that involves a mystery — it doe...\n", "66629 Write a story that involves a mystery — it doe...\n", "66630 Write a story that involves a mystery — it doe...\n", " ... \n", "66832 Write a story that involves a mystery — it doe...\n", "66833 Write a story that involves a mystery — it doe...\n", "66834 Write a story that involves a mystery — it doe...\n", "66835 Write a story that involves a mystery — it doe...\n", "66836 Write a story that involves a mystery — it doe...\n", "Name: prompt, Length: 211, dtype: object\n", "prompt_1025\n", "47034 Write about someone who gets stuck in their wo...\n", "47035 Write about someone who gets stuck in their wo...\n", "47036 Write about someone who gets stuck in their wo...\n", "47037 Write about someone who gets stuck in their wo...\n", "47038 Write about someone who gets stuck in their wo...\n", " ... \n", "47193 Write about someone who gets stuck in their wo...\n", "47194 Write about someone who gets stuck in their wo...\n", "47195 Write about someone who gets stuck in their wo...\n", "47196 Write about someone who gets stuck in their wo...\n", "47197 Write about someone who gets stuck in their wo...\n", "Name: prompt, Length: 164, dtype: object\n", "prompt_1157\n", "67225 Write a story that begins and ends with someon...\n", "67226 Write a story that begins and ends with someon...\n", "67227 Write a story that begins and ends with someon...\n", "67228 Write a story that begins and ends with someon...\n", "67229 Write a story that begins and ends with someon...\n", " ... \n", "67629 Write a story that begins and ends with someon...\n", "67630 Write a story that begins and ends with someon...\n", "67631 Write a story that begins and ends with someon...\n", "67632 Write a story that begins and ends with someon...\n", "67633 Write a story that begins and ends with someon...\n", "Name: prompt, Length: 409, dtype: object\n", "prompt_0992\n", "41767 Write a story that spans exactly a year and ta...\n", "41768 Write a story that spans exactly a year and ta...\n", "41769 Write a story that spans exactly a year and ta...\n", "41770 Write a story that spans exactly a year and ta...\n", "41771 Write a story that spans exactly a year and ta...\n", " ... \n", "41974 Write a story that spans exactly a year and ta...\n", "41975 Write a story that spans exactly a year and ta...\n", "41976 Write a story that spans exactly a year and ta...\n", "41977 Write a story that spans exactly a year and ta...\n", "41978 Write a story that spans exactly a year and ta...\n", "Name: prompt, Length: 212, dtype: object\n", "prompt_0118\n", "6222 Write a story about someone trying to reinvent...\n", "6223 Write a story about someone trying to reinvent...\n", "6224 Write a story about someone trying to reinvent...\n", "6225 Write a story about someone trying to reinvent...\n", "6226 Write a story about someone trying to reinvent...\n", "6227 Write a story about someone trying to reinvent...\n", "6228 Write a story about someone trying to reinvent...\n", "6229 Write a story about someone trying to reinvent...\n", "6230 Write a story about someone trying to reinvent...\n", "6231 Write a story about someone trying to reinvent...\n", "6232 Write a story about someone trying to reinvent...\n", "6233 Write a story about someone trying to reinvent...\n", "6234 Write a story about someone trying to reinvent...\n", "6235 Write a story about someone trying to reinvent...\n", "6236 Write a story about someone trying to reinvent...\n", "6237 Write a story about someone trying to reinvent...\n", "6238 Write a story about someone trying to reinvent...\n", "6239 Write a story about someone trying to reinvent...\n", "6240 Write a story about someone trying to reinvent...\n", "6241 Write a story about someone trying to reinvent...\n", "6242 Write a story about someone trying to reinvent...\n", "6243 Write a story about someone trying to reinvent...\n", "6244 Write a story about someone trying to reinvent...\n", "6245 Write a story about someone trying to reinvent...\n", "6246 Write a story about someone trying to reinvent...\n", "6247 Write a story about someone trying to reinvent...\n", "6248 Write a story about someone trying to reinvent...\n", "6249 Write a story about someone trying to reinvent...\n", "6250 Write a story about someone trying to reinvent...\n", "6251 Write a story about someone trying to reinvent...\n", "6252 Write a story about someone trying to reinvent...\n", "6253 Write a story about someone trying to reinvent...\n", "6254 Write a story about someone trying to reinvent...\n", "6255 Write a story about someone trying to reinvent...\n", "6256 Write a story about someone trying to reinvent...\n", "6257 Write a story about someone trying to reinvent...\n", "6258 Write a story about someone trying to reinvent...\n", "6259 Write a story about someone trying to reinvent...\n", "Name: prompt, dtype: object\n", "prompt_1162\n", "68371 Write a story about a proposal. \n", "68372 Write a story about a proposal. \n", "68373 Write a story about a proposal. \n", "68374 Write a story about a proposal. \n", "68375 Write a story about a proposal. \n", " ... \n", "68580 Write a story about a proposal. \n", "68581 Write a story about a proposal. \n", "68582 Write a story about a proposal. \n", "68583 Write a story about a proposal. \n", "68584 Write a story about a proposal. \n", "Name: prompt, Length: 214, dtype: object\n", "prompt_1172\n", "69960 Write about someone who has a superpower.\n", "69961 Write about someone who has a superpower.\n", "69962 Write about someone who has a superpower.\n", "69963 Write about someone who has a superpower.\n", "69964 Write about someone who has a superpower.\n", " ... \n", "70292 Write about someone who has a superpower.\n", "70293 Write about someone who has a superpower.\n", "70294 Write about someone who has a superpower.\n", "70295 Write about someone who has a superpower.\n", "70296 Write about someone who has a superpower.\n", "Name: prompt, Length: 337, dtype: object\n", "the first example of train is prompt_id prompt_1234\n", "story1_id rikt93\n", "story2_id qyd9jh\n", "time_lag 309660.0\n", "least_likes 8\n", "chosen_text <|im_start|>user\\nWrite a story about som...\n", "rejected_text <|im_start|>user\\nWrite a story about som...\n", "Name: 0, dtype: object\n" ] } ], "source": [ "#test dataloader \n", "from dataloader import StoryPairDataset\n", "dataloader = StoryPairDataset(datapath,\n", " pairpath,\n", " tokenizer,\n", " task='rm',\n", " used_dataset_size=100,\n", " train_test_split=0.1,\n", " split_by='random',\n", " max_len=1024*4,\n", " mode='m2',\n", " max_time_window=5400,\n", " least_likes=5,\n", " margin=False)" ] }, { "cell_type": "code", "execution_count": null, "id": "b11f608a-c2eb-42af-bfba-e801ee40e0ed", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "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.10.13" } }, "nbformat": 4, "nbformat_minor": 5 }