hmm-rhyme-and-meter.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "from __future__ import division, print_function\n",
    "import numpy as np\n",
    "from hmmlearn.hmm import MultinomialHMM\n",
    "import poetrytools\n",
    "import itertools"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "%run 'preprocessing.ipynb'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": true,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "def get_rhymes(words):\n",
    "    num_words = len(words)\n",
    "    rhymes = np.empty((num_words, num_words))\n",
    "    \n",
    "    for i in range(num_words):\n",
    "        for j in (k for k in range(num_words) if not k == i):\n",
    "            if poetrytools.rhymes(words[i], words[j]):\n",
    "                rhymes[i][j] = 1\n",
    "    return rhymes"
   ]
  },
  {
   "cell_type": "code",
<<<<<<< HEAD
   "execution_count": 1,
=======
   "execution_count": 4,
>>>>>>> a93e95d13aae8ad2b6f95257e9facec7601735b0
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "quatrain_rhymes = get_rhymes(quatrain_words)\n",
    "volta_rhymes = get_rhymes(volta_words)\n",
    "couplet_rhymes = get_rhymes(couplet_words)"
   ]
  },
  {
   "cell_type": "code",
<<<<<<< HEAD
   "execution_count": 2,
=======
   "execution_count": 35,
>>>>>>> a93e95d13aae8ad2b6f95257e9facec7601735b0
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
<<<<<<< HEAD
   "outputs": [],
=======
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "         1      -76127.4587             +nan\n",
      "         2      -62169.0164      +13958.4424\n",
      "         3      -62161.6852          +7.3311\n",
      "         4      -62139.8518         +21.8334\n",
      "         5      -62075.8032         +64.0485\n",
      "         6      -61903.4710        +172.3323\n",
      "         7      -61525.6623        +377.8087\n",
      "         8      -60947.1629        +578.4994\n",
      "         9      -60391.2399        +555.9230\n",
      "        10      -60054.9712        +336.2688\n",
      "        11      -59898.7616        +156.2096\n",
      "        12      -59814.0838         +84.6778\n",
      "        13      -59734.7823         +79.3015\n",
      "        14      -59623.9225        +110.8598\n",
      "        15      -59458.5824        +165.3401\n",
      "        16      -59238.7651        +219.8173\n",
      "        17      -59001.2101        +237.5549\n",
      "        18      -58790.0781        +211.1321\n",
      "        19      -58605.5207        +184.5574\n",
      "        20      -58404.3310        +201.1897\n",
      "        21      -58134.2876        +270.0433\n",
      "        22      -57753.6637        +380.6239\n",
      "        23      -57248.0123        +505.6514\n",
      "        24      -56652.2620        +595.7504\n",
      "        25      -56030.7322        +621.5298\n",
      "        26      -55433.8485        +596.8837\n",
      "        27      -54885.7186        +548.1299\n",
      "        28      -54391.9014        +493.8172\n",
      "        29      -53922.4361        +469.4653\n",
      "        30      -53469.0739        +453.3621\n",
      "        31      -53040.1752        +428.8988\n",
      "        32      -52650.5704        +389.6048\n",
      "        33      -52303.6336        +346.9368\n",
      "        34      -52006.6988        +296.9348\n",
      "        35      -51763.5944        +243.1044\n",
      "        36      -51572.5215        +191.0729\n",
      "        37      -51420.7029        +151.8186\n",
      "        38      -51300.2478        +120.4551\n",
      "        39      -51208.7231         +91.5247\n",
      "        40      -51141.7626         +66.9605\n",
      "        41      -51086.7961         +54.9665\n",
      "        42      -51040.8907         +45.9053\n",
      "        43      -51003.2558         +37.6350\n",
      "        44      -50971.3685         +31.8873\n",
      "        45      -50945.6952         +25.6732\n",
      "        46      -50923.4293         +22.2659\n",
      "        47      -50903.6356         +19.7937\n",
      "        48      -50887.3729         +16.2626\n",
      "        49      -50872.4153         +14.9576\n",
      "        50      -50859.1431         +13.2721\n",
      "         1      -35724.9265             +nan\n",
      "         2      -30165.9506       +5558.9759\n",
      "         3      -30162.0418          +3.9088\n",
      "         4      -30149.7789         +12.2629\n",
      "         5      -30111.6296         +38.1493\n",
      "         6      -30004.2666        +107.3630\n",
      "         7      -29773.4420        +230.8246\n",
      "         8      -29455.9543        +317.4877\n",
      "         9      -29183.9047        +272.0496\n",
      "        10      -29024.0935        +159.8113\n",
      "        11      -28945.6056         +78.4879\n",
      "        12      -28894.6316         +50.9740\n",
      "        13      -28834.4421         +60.1895\n",
      "        14      -28742.1649         +92.2772\n",
      "        15      -28613.4574        +128.7075\n",
      "        16      -28470.8124        +142.6450\n",
      "        17      -28347.0959        +123.7165\n",
      "        18      -28248.8547         +98.2412\n",
      "        19      -28152.6424         +96.2123\n",
      "        20      -28028.4077        +124.2346\n",
      "        21      -27850.0006        +178.4072\n",
      "        22      -27603.1291        +246.8714\n",
      "        23      -27293.1398        +309.9893\n",
      "        24      -26941.4170        +351.7228\n",
      "        25      -26572.0795        +369.3375\n",
      "        26      -26199.9036        +372.1759\n",
      "        27      -25833.1807        +366.7229\n",
      "        28      -25478.9762        +354.2045\n",
      "        29      -25151.8603        +327.1159\n",
      "        30      -24868.7703        +283.0900\n",
      "        31      -24639.9461        +228.8242\n",
      "        32      -24454.0277        +185.9184\n",
      "        33      -24308.3626        +145.6651\n",
      "        34      -24193.3010        +115.0616\n",
      "        35      -24100.5333         +92.7677\n",
      "        36      -24030.4478         +70.0855\n",
      "        37      -23973.7767         +56.6711\n",
      "        38      -23930.0641         +43.7126\n",
      "        39      -23895.5623         +34.5018\n",
      "        40      -23861.4921         +34.0702\n",
      "        41      -23832.9384         +28.5537\n",
      "        42      -23812.8627         +20.0758\n",
      "        43      -23794.2189         +18.6438\n",
      "        44      -23778.5137         +15.7052\n",
      "        45      -23766.9865         +11.5272\n",
      "        46      -23757.3485          +9.6379\n",
      "        47      -23750.1874          +7.1612\n",
      "        48      -23742.4817          +7.7057\n",
      "        49      -23735.5376          +6.9441\n",
      "        50      -23729.5954          +5.9422\n",
      "         1      -17914.6424             +nan\n",
      "         2      -15391.9056       +2522.7368\n",
      "         3      -15388.7817          +3.1239\n",
      "         4      -15379.6818          +9.0999\n",
      "         5      -15354.2037         +25.4781\n",
      "         6      -15291.3098         +62.8939\n",
      "         7      -15167.3734        +123.9364\n",
      "         8      -15001.2055        +166.1679\n",
      "         9      -14873.7036        +127.5020\n",
      "        10      -14810.6400         +63.0636\n",
      "        11      -14776.9708         +33.6692\n",
      "        12      -14747.2699         +29.7009\n",
      "        13      -14706.2568         +41.0131\n",
      "        14      -14643.7621         +62.4947\n",
      "        15      -14564.9012         +78.8609\n",
      "        16      -14491.3483         +73.5530\n",
      "        17      -14432.4409         +58.9073\n",
      "        18      -14375.0837         +57.3573\n",
      "        19      -14300.2626         +74.8211\n",
      "        20      -14191.9036        +108.3590\n",
      "        21      -14043.5509        +148.3527\n",
      "        22      -13862.6123        +180.9386\n",
      "        23      -13662.1460        +200.4663\n",
      "        24      -13445.7940        +216.3520\n",
      "        25      -13214.7952        +230.9988\n",
      "        26      -12981.4216        +233.3736\n",
      "        27      -12766.0324        +215.3892\n",
      "        28      -12575.3459        +190.6866\n",
      "        29      -12417.0590        +158.2868\n",
      "        30      -12295.2112        +121.8478\n",
      "        31      -12198.4228         +96.7884\n",
      "        32      -12131.7963         +66.6265\n",
      "        33      -12085.9741         +45.8222\n",
      "        34      -12052.6202         +33.3539\n",
      "        35      -12027.9020         +24.7182\n",
      "        36      -12008.9803         +18.9217\n",
      "        37      -11994.8851         +14.0952\n",
      "        38      -11982.2660         +12.6191\n",
      "        39      -11968.9670         +13.2990\n",
      "        40      -11955.5440         +13.4231\n",
      "        41      -11943.1565         +12.3874\n",
      "        42      -11933.2075          +9.9491\n",
      "        43      -11926.8468          +6.3606\n",
      "        44      -11921.7404          +5.1064\n",
      "        45      -11917.4362          +4.3042\n",
      "        46      -11913.5633          +3.8729\n",
      "        47      -11910.0915          +3.4718\n",
      "        48      -11906.5379          +3.5535\n",
      "        49      -11903.0652          +3.4728\n",
      "        50      -11900.7938          +2.2714\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "MultinomialHMM(algorithm='viterbi', init_params='ste', n_components=30,\n",
       "        n_iter=50, params='ste',\n",
       "        random_state=<mtrand.RandomState object at 0x111e745a0>,\n",
       "        startprob_prior=1.0, tol=0.01, transmat_prior=1.0, verbose=True)"
      ]
     },
     "execution_count": 35,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
>>>>>>> a93e95d13aae8ad2b6f95257e9facec7601735b0
   "source": [
    "reversed_quatrain_model = MultinomialHMM(n_components=30, n_iter=50, verbose=True)\n",
    "reversed_volta_model = MultinomialHMM(n_components=30, n_iter=50, verbose=True)\n",
    "reversed_couplet_model = MultinomialHMM(n_components=30, n_iter=50, verbose=True)\n",
    "reversed_quatrain_model.fit(np.concatenate(reversed_converted_quatrain_lines), quatrain_lengths)\n",
    "reversed_volta_model.fit(np.concatenate(reversed_converted_volta_lines), volta_lengths)\n",
    "reversed_couplet_model.fit(np.concatenate(reversed_converted_couplet_lines), couplet_lengths)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "def generate_line(model, words, seed=None, rhymes=None):\n",
    "    syllables_left = 10\n",
    "    last_stress = 0\n",
    "    \n",
    "    line= []\n",
    "    start_probs = model.startprob_\n",
    "    emission_probs = model.emissionprob_\n",
    "    transition_probs = model.transmat_\n",
    "    \n",
    "    start_state = np.random.choice(len(start_probs), p = start_probs)\n",
    "    \n",
    "    if seed is not None:\n",
    "        possible_start_emissions = np.where(rhymes[seed] == 1)\n",
    "        probs = np.array(emission_probs[start_state][possible_start_emissions])\n",
    "\n",
    "        scaled_probs = probs / sum(probs)\n",
    "        while True:\n",
    "            start_emission = np.random.choice(possible_start_emissions[0], p=scaled_probs)\n",
    "            start_stress = poetrytools.stress(words[start_emission])\n",
    "            if len(start_stress) == 1 or int(start_stress[-1]) == 1 :\n",
    "                break\n",
    "    else:\n",
    "        while True:\n",
    "            start_emission = np.random.choice(len(emission_probs[start_state]), p=emission_probs[start_state])\n",
    "            start_stress = poetrytools.stress(words[start_emission])\n",
    "            if len(start_stress) == 1 or int(start_stress[-1]) == 1:\n",
    "                break\n",
    "    \n",
    "    line.append(start_emission)\n",
    "    start_stress = poetrytools.stress(words[start_emission])\n",
    "    syllables_left -= len(start_stress)\n",
    "    \n",
    "    if len(start_stress) == 1:\n",
    "        prev_starting_stress = 1\n",
    "    else:\n",
    "        prev_starting_stress = int(start_stress[0])\n",
    "\n",
    "    curr_state = start_state\n",
    "    while syllables_left > 0:\n",
    "        possible_transitions = transition_probs[curr_state]\n",
    "        curr_state = np.random.choice(len(possible_transitions), p=possible_transitions)\n",
    "        possible_emissions = emission_probs[curr_state]\n",
    "        while True:\n",
    "            curr_emission = np.random.choice(len(possible_emissions), p=possible_emissions)\n",
    "            curr_stress = poetrytools.stress(words[curr_emission])\n",
    "            if len(curr_stress) == 1:\n",
    "                prev_starting_stress = 1 - prev_starting_stress\n",
    "                syllables_left -= 1\n",
    "                break\n",
    "            elif len(curr_stress) > syllables_left or int(curr_stress[-1]) == prev_starting_stress:\n",
    "                continue\n",
    "            else:\n",
    "                prev_starting_stress = int(curr_stress[0])\n",
    "                syllables_left -= len(curr_stress)\n",
    "                break\n",
    "        line.append(curr_emission)\n",
    "\n",
    "    return line\n",
    "\n",
    "def convert_line(sample, words):\n",
    "    ret = ''\n",
    "    i = 0\n",
    "    for word in reversed(sample):\n",
    "        curr_word = words[word]\n",
    "        if i == 0 or (curr_word == 'i'):\n",
    "            ret += curr_word.title() + ' '\n",
    "        else:\n",
    "            ret += curr_word + ' '\n",
    "        i += 1\n",
    "    return ret\n",
    "\n",
    "def generate_pair(model, words, rhymes):\n",
    "    while True:\n",
    "        a_line = generate_line(model, words)\n",
    "        seed = a_line[0]\n",
    "        if len(np.where(rhymes[seed] == 1)[0]) > 0:\n",
    "            b_line = generate_line(model, words, seed, rhymes)\n",
    "            return a_line, b_line\n",
    "        \n",
    "def generate_rhyming_and_meter_sonnet():\n",
    "    sonnet = ''\n",
    "    a_lines = []\n",
    "    b_lines = []\n",
    "    \n",
    "    for _ in range(4):\n",
    "        a_line, b_line = generate_pair(reversed_quatrain_model, quatrain_words, quatrain_rhymes)\n",
    "        a_lines.append(a_line)\n",
    "        b_lines.append(b_line)\n",
    "    \n",
    "    for i in range(2):\n",
    "        sonnet += convert_line(a_lines[2 * i], quatrain_words) + '\\n'\n",
    "        sonnet += convert_line(a_lines[2 * i + 1], quatrain_words) + '\\n'\n",
    "        sonnet += convert_line(b_lines[2 * i], quatrain_words) + '\\n'\n",
    "        sonnet += convert_line(b_lines[2 * i + 1], quatrain_words) + '\\n'\n",
    "    \n",
    "    a_lines = []\n",
    "    b_lines = []\n",
    "    \n",
    "    for _ in range(2):\n",
    "        a_line, b_line = generate_pair(reversed_volta_model, volta_words, volta_rhymes)\n",
    "        a_lines.append(a_line)\n",
    "        b_lines.append(b_line)\n",
    "    \n",
    "    sonnet += convert_line(a_lines[0], volta_words) + '\\n'\n",
    "    sonnet += convert_line(a_lines[1], volta_words) + '\\n'\n",
    "    sonnet += convert_line(b_lines[0], volta_words) + '\\n'\n",
    "    sonnet += convert_line(b_lines[1], volta_words) + '\\n'\n",
    "    \n",
    "    a_line, b_line = generate_pair(reversed_couplet_model, couplet_words, couplet_rhymes)\n",
    "    sonnet += convert_line(a_line, couplet_words) + '\\n'\n",
    "    sonnet += convert_line(b_line, couplet_words) + '\\n'\n",
    "    \n",
    "    return sonnet\n",
    "\n",
    "def generate_10_rhyming_and_meter_sonnets():\n",
    "    sonnets = ''\n",
    "    for i in range(10):\n",
    "        print('Generating Sonnet ' + str(i + 1))\n",
    "        sonnets += str(i) + '\\n' + generate_rhyming_and_meter_sonnet() + '\\n'\n",
    "    \n",
    "    f = open(\"project2data/rhyming_and_meter_shakespeare.txt\",\"w\")\n",
    "    f.write(sonnets)\n",
    "    return sonnets"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Generating Sonnet1\n",
      "Generating Sonnet2\n",
      "Generating Sonnet3\n",
      "Generating Sonnet4\n",
      "Generating Sonnet5\n",
      "Generating Sonnet6\n",
      "Generating Sonnet7\n",
      "Generating Sonnet8\n",
      "Generating Sonnet9\n",
      "Generating Sonnet10\n",
      "0\n",
      "My speed hath all filled and thy shouldst redeem \n",
      "Nor my rose of a height that far have year \n",
      "For see more should mock me thing and the deem \n",
      "Such the bear though their time when since your near \n",
      "Smell thoughts my odour of life worth's her heart \n",
      "Pen the darkening will your face to you \n",
      "From fingers me my able as resort \n",
      "Leads my rose to every siren due \n",
      "Crowning strong blessed at is truetelling strong \n",
      "Have guess that me from base look eyes my doom \n",
      "Those beauty's say dissuade on my belong \n",
      "Sworn to triumphant and one straying room \n",
      "That then me keep but virtue thy to heart \n",
      "Pays sing for live else suns my love in sort \n",
      "\n",
      "1\n",
      "Boughs I that sweet least might you leaves redeem \n",
      "And for I breath such self of his excess \n",
      "That sick forbid are with give did the deem \n",
      "Or which knows you than see did it we press \n",
      "And love for if that I why for friends beck \n",
      "Therefore from a own my love gives our stars \n",
      "Of heaven self that appetite and check \n",
      "Than at are doubting your he to that tears \n",
      "That I I thee thee mine my even fruit \n",
      "Own and therefore of my directly pride \n",
      "It having when in my hand or is mute \n",
      "For what but seest his marvel but wolf hide \n",
      "And day turn daily men with dying days \n",
      "Live who for life me looks I robs fair praise \n",
      "\n",
      "2\n",
      "Thy now and are is life her high should fears \n",
      "Pilgrimage me part can what heaven verse \n",
      "Thou for upon that humour is the tears \n",
      "Thou sweet whereto once on the disperse \n",
      "To none all brief in speak against the thing \n",
      "Hath be a making to end down nor doom \n",
      "Love is you the still is antique spring \n",
      "Shall lovely blunt for what be I are womb \n",
      "Great own be break as blanks to winter's night \n",
      "As star and others' he which from years place \n",
      "Poor buried calls waste in to winter's quite \n",
      "May gav'st fair more what proud be beauty case \n",
      "Death eye's draw then and king so for seem mind \n",
      "Have the love nightly still and night I find \n",
      "\n",
      "3\n",
      "And trouble those it on to eye was short \n",
      "Canst for when do ruth what thou sooner know \n",
      "Of was not my mind's than as evil art \n",
      "Lo as the ugly I in their vile no \n",
      "When with your now and the love were my rite \n",
      "Wish so but in for mine deeds mine my there \n",
      "Sin towers shalt and which that pen of night \n",
      "Thee it me losing if pays my am rare \n",
      "Say do retention they o though you grow \n",
      "That I love me this wonder in in moan \n",
      "To heart could it than the fair in shall throw \n",
      "Wilt prove the eyes she victors by was groan \n",
      "Here thou his so shall need she by all heart \n",
      "What'S old will since heats I be of report \n",
      "\n",
      "4\n",
      "Eye with welfare love seem my body's bold \n",
      "With thee year weary whilst to scanted face \n",
      "Could be alleating death's doth things assailed \n",
      "Thy earth with I argument do disgrace \n",
      "Being outlive upon I love in know \n",
      "On can not sweet memory of expense \n",
      "Snow that clear change sensual not from no \n",
      "Strength'S to ripe where worst to worth of the thence \n",
      "Time but thy pace god to prepare thy sight \n",
      "In him that live if art what out true heart \n",
      "Proud that I silence hath first thou self write \n",
      "With art o let I early bring was art \n",
      "That he on only cunning name thee care \n",
      "I summer's drugs o I son dost compare \n",
      "\n",
      "5\n",
      "To say that virtue untutored fill \n",
      "Far it confine love bound struck the contend \n",
      "My thy receipt and flame to bright did skill \n",
      "Mars thou still than damasked bright of stand \n",
      "A heart and than as hungry must the find \n",
      "Art cheek by health her view with shade am cure \n",
      "What comment her mouthed death I and my crowned \n",
      "I grant my scope to love plague I endure \n",
      "Sometime time's thee with I born mistress' those \n",
      "In had nine would in nor sometime to more \n",
      "So heaven captive canst fame words suppose \n",
      "Points we remembrance from of gav'st a store \n",
      "Thee what is in old is that add whit brings \n",
      "Joy such since for since have and none be kings \n",
      "\n",
      "6\n",
      "If I and eye and I watery stout \n",
      "If said I 'tis that clouds doth breasts to part \n",
      "Every own so music winters out \n",
      "In rich and oft what I for he and heart \n",
      "Then true I by the I the prey my use \n",
      "A win I beauty was though where that best \n",
      "These me to windy blind do days abuse \n",
      "When that I which not I the eye to most \n",
      "A save and from fears how with gentle use \n",
      "What thou age thy make since look times thou youth \n",
      "In save walk heaven to thoughts then abuse \n",
      "I take though they not this is being truth \n",
      "Thy heart thou he in this I with report \n",
      "This both thy you or can thou of to sort \n",
      "\n",
      "7\n",
      "Faults I and eye can speak of lusty heart \n",
      "No all what to our grace every grief \n",
      "His picture is me thou far he called art \n",
      "What flower words disgrace a double thief \n",
      "Most raiment trees so hands though some men hue \n",
      "Dumb eat when ugly true I all my due \n",
      "But the your tired that's from nature's view \n",
      "Time'S friend taught us summer's throned review \n",
      "Till fairer in love come be be expressed \n",
      "Nor give and in from envy this belong \n",
      "Termed the eyes of love's others better haste \n",
      "So but this no love not of I so strong \n",
      "Scorn not this where this and live let in lies \n",
      "Kill with you but live who not former eyes \n",
      "\n",
      "8\n",
      "Great name of credit being fierce did time \n",
      "Sweet public I wink you how where plea rare \n",
      "Who that me not is my wealth lap more prime \n",
      "But wit me o since in till is despair \n",
      "What left the mine form with your shalt with come \n",
      "Enjoyed with me she and foot not suppressed \n",
      "Thy summer's he to say of little sum \n",
      "Doth sail of such least all expense doth trust \n",
      "Like thou sea serving were accumulate \n",
      "Mad nor heavily she not so I last \n",
      "Those harsh but on have doth anticipate \n",
      "Write this self his blood had contracted trust \n",
      "Of heart's my state mine being her dost still \n",
      "Must thine be says the have yet him my ill \n",
      "\n",
      "9\n",
      "It where compare that in my time endured \n",
      "Oft that my make me soul of thou confound \n",
      "Same for may born that some confined expired \n",
      "Deaths do old your self of my rose no wind \n",
      "Year thou faults and we an called discontent \n",
      "The worth adore vice the but the you near \n",
      "A nobler linger breath and bright surmount \n",
      "Those lived hath to thy signs wide for things tear \n",
      "Speak and by thy looks to controlling give \n",
      "Taught very resty on her must betray \n",
      "Wrecked how her both when my come have my live \n",
      "Other neither in my offence's prey \n",
      "Thus for ne'er looks thou die of leave thy song \n",
      "My love thy this and mourn who hath rude long \n",
      "\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/usr/local/lib/python2.7/site-packages/ipykernel/__main__.py:16: RuntimeWarning: invalid value encountered in true_divide\n",
      "/usr/local/lib/python2.7/site-packages/ipykernel/__main__.py:18: RuntimeWarning: invalid value encountered in less\n"
     ]
    }
   ],
   "source": [
    "print(generate_10_rhyming_and_meter_sonnets())"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 141,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": [
    "# checks if line is in iambic pentameter (i.e. 0101010101 stress pattern)\n",
    "def check_iambic_pentameter(line):\n",
    "    # get the stresses from cmu dict \n",
    "    # if word is 1 syllable, then have the option for it to be stressed or unstressed\n",
    "    stresses = []\n",
    "    for i in line.split(' '):\n",
    "        stress = poetrytools.stress(i)\n",
    "        if len(stress) == 1:\n",
    "            stresses.append(['0','1'])\n",
    "        else:\n",
    "            stresses.append([stress])\n",
    "    \n",
    "    # make combination of all possible stress patterns\n",
    "    result = [[]]\n",
    "    final = []\n",
    "    for pool in stresses:\n",
    "        result = [x+[y] for x in result for y in pool]\n",
    "    final = [''.join(i) for i in result]\n",
    "    \n",
    "    # return if any pattern fits iambic pentameter \n",
    "    return ('0101010101' in final)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true,
    "deletable": true,
    "editable": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}