test_control.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from predpatt import PredPatt, PredPattOpts\n",
    "from predpatt import load_conllu"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "prevent_example = \"\"\"\n",
    "1\tOne\tone\tNUM\tCD\tNumType=Card\t2\tnummod\t_\t_\n",
    "2\tanswer\tanswer\tNOUN\tNN\tNumber=Sing\t3\tnsubj\t_\t_\n",
    "3\tis\tbe\tVERB\tVBZ\tMood=Ind|Number=Sing|Person=3|Tense=Pres|VerbForm=Fin\t0\troot\t_\t_\n",
    "4\tthat\tthat\tSCONJ\tIN\t_\t7\tmark\t_\t_\n",
    "5\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t6\tdet\t_\t_\n",
    "6\tPentagon\tPentagon\tPROPN\tNNP\tNumber=Sing\t7\tnsubj\t_\t_\n",
    "7\tprevented\tprevent\tVERB\tVBD\tMood=Ind|Tense=Past|VerbForm=Fin\t3\tccomp\t_\t_\n",
    "8\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t10\tdet\t_\t_\n",
    "9\tState\tState\tPROPN\tNNP\tNumber=Sing\t10\tcompound\t_\t_\n",
    "10\tDepartment\tDepartment\tPROPN\tNNP\tNumber=Sing\t7\tdobj\t_\t_\n",
    "11\tfrom\tfrom\tSCONJ\tIN\t_\t12\tmark\t_\t_\n",
    "12\trunning\trun\tVERB\tVBG\tVerbForm=Ger\t7\tadvcl\t_\t_\n",
    "13\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t14\tdet\t_\t_\n",
    "14\tCPA\tCPA\tPROPN\tNNP\tNumber=Sing\t12\tdobj\t_\tSpaceAfter=No\n",
    "15\t.\t.\tPUNCT\t.\t_\t3\tpunct\t_\t_\n",
    "\"\"\"\n",
    "\n",
    "delay_example = \"\"\"\n",
    "1\tThe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t2\tdet\t_\t_\n",
    "2\tjudge\tjudge\tNOUN\tNN\tNumber=Sing\t4\tnsubj\t_\t_\n",
    "3\thad\thave\tAUX\tVBD\tMood=Ind|Tense=Past|VerbForm=Fin\t4\taux\t_\t_\n",
    "4\tagreed\tagree\tVERB\tVBN\tTense=Past|VerbForm=Part\t0\troot\t_\t_\n",
    "5\tto\tto\tPART\tTO\t_\t6\tmark\t_\t_\n",
    "6\tdelay\tdelay\tVERB\tVB\tVerbForm=Inf\t4\txcomp\t_\t_\n",
    "7\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t10\tdet\t_\t_\n",
    "8\tcivil\tcivil\tADJ\tJJ\tDegree=Pos\t9\tamod\t_\t_\n",
    "9\trights\trights\tNOUN\tNNS\tNumber=Plur\t10\tcompound\t_\t_\n",
    "10\tmatter\tmatter\tNOUN\tNN\tNumber=Sing\t6\tdobj\t_\t_\n",
    "11\tfrom\tfrom\tADP\tIN\t_\t12\tcase\t_\t_\n",
    "12\tproceeding\tproceeding\tNOUN\tNN\tNumber=Sing\t6\tnmod\t_\t_\n",
    "13\tuntil\tuntil\tADP\tIN\t_\t16\tcase\t_\t_\n",
    "14\tat\tat\tADV\tRB\t_\t15\tcase\t_\t_\n",
    "15\tleast\tleast\tADV\tRBS\tDegree=Sup\t16\tnmod\t_\t_\n",
    "16\tOctober\tOctober\tPROPN\tNNP\tNumber=Sing\t6\tnmod\t_\t_\n",
    "17\t2004\t2004\tNUM\tCD\tNumType=Card\t16\tnummod\t_\tSpaceAfter=No\n",
    "18\t.\t.\tPUNCT\t.\t_\t4\tpunct\t_\t_\n",
    "\"\"\"\n",
    "\n",
    "discourage_example = \"\"\"\n",
    "1\tSince\tsince\tADP\tIN\t_\t2\tcase\t_\t_\n",
    "2\treintroduction\treintroduction\tNOUN\tNN\tNumber=Sing\t10\tnmod\t_\tSpaceAfter=No\n",
    "3\t,\t,\tPUNCT\t,\t_\t10\tpunct\t_\t_\n",
    "4\thowever\thowever\tADV\tRB\t_\t10\tadvmod\t_\tSpaceAfter=No\n",
    "5\t,\t,\tPUNCT\t,\t_\t10\tpunct\t_\t_\n",
    "6\tRipple\tRipple\tPROPN\tNNP\tNumber=Sing\t10\tnsubj\t_\t_\n",
    "7\tand\tand\tCONJ\tCC\t_\t6\tcc\t_\t_\n",
    "8\tBeschta\tBeschta\tPROPN\tNNP\tNumber=Sing\t6\tconj\t_\t_\n",
    "9\thave\thave\tAUX\tVBP\tMood=Ind|Tense=Pres|VerbForm=Fin\t10\taux\t_\t_\n",
    "10\tfound\tfind\tVERB\tVBN\tTense=Past|VerbForm=Part\t0\troot\t_\tSpaceAfter=No\n",
    "11\t,\t,\tPUNCT\t,\t_\t10\tpunct\t_\t_\n",
    "12\tfear\tfear\tNOUN\tNN\tNumber=Sing\t18\tnsubj\t_\t_\n",
    "13\tof\tof\tADP\tIN\t_\t15\tcase\t_\t_\n",
    "14\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t15\tdet\t_\t_\n",
    "15\twolves\twolf\tNOUN\tNNS\tNumber=Plur\t12\tnmod\t_\t_\n",
    "16\tapparently\tapparently\tADV\tRB\t_\t18\tadvmod\t_\t_\n",
    "17\thas\thave\tAUX\tVBZ\tMood=Ind|Number=Sing|Person=3|Tense=Pres|VerbForm=Fin\t18\taux\t_\t_\n",
    "18\tdiscouraged\tdiscourage\tVERB\tVBN\tTense=Past|VerbForm=Part\t10\tccomp\t_\t_\n",
    "19\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t20\tdet\t_\t_\n",
    "20\telk\telk\tNOUN\tNNS\tNumber=Plur\t18\tdobj\t_\t_\n",
    "21\tfrom\tfrom\tSCONJ\tIN\t_\t22\tmark\t_\t_\n",
    "22\tspending\tspend\tVERB\tVBG\tVerbForm=Ger\t18\tadvcl\t_\t_\n",
    "23\ttoo\ttoo\tADV\tRB\t_\t24\tadvmod\t_\t_\n",
    "24\tmuch\tmuch\tADJ\tJJ\tDegree=Pos\t25\tamod\t_\t_\n",
    "25\ttime\ttime\tNOUN\tNN\tNumber=Sing\t22\tdobj\t_\t_\n",
    "26\tat\tat\tADP\tIN\t_\t28\tcase\t_\t_\n",
    "27\tstream\tstream\tNOUN\tNN\tNumber=Sing\t28\tcompound\t_\t_\n",
    "28\tbanks\tbank\tNOUN\tNNS\tNumber=Plur\t22\tnmod\t_\t_\n",
    "29\t--\t--\tPUNCT\t,\t_\t28\tpunct\t_\t_\n",
    "30\twhere\twhere\tADV\tWRB\tPronType=Rel\t33\tadvmod\t_\t_\n",
    "31\tthey\tthey\tPRON\tPRP\tCase=Nom|Number=Plur|Person=3|PronType=Prs\t33\tnsubj\t_\t_\n",
    "32\tare\tbe\tVERB\tVBP\tMood=Ind|Tense=Pres|VerbForm=Fin\t33\tcop\t_\t_\n",
    "33\tout\tout\tADV\tRB\t_\t28\tacl:relcl\t_\t_\n",
    "34\tin\tin\tADP\tIN\t_\t36\tcase\t_\t_\n",
    "35\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t36\tdet\t_\t_\n",
    "36\topen\topen\tADJ\tJJ\tDegree=Pos\t33\tnmod\t_\t_\n",
    "37\t--\t--\tPUNCT\t,\t_\t22\tpunct\t_\t_\n",
    "38\tand\tand\tCONJ\tCC\t_\t22\tcc\t_\t_\n",
    "39\tmunching\tmunch\tVERB\tVBG\tVerbForm=Ger\t22\tconj\t_\t_\n",
    "40\ton\ton\tADP\tIN\t_\t42\tcase\t_\t_\n",
    "41\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t42\tdet\t_\t_\n",
    "42\tsaplings\tsapling\tNOUN\tNNS\tNumber=Plur\t39\tnmod\t_\tSpaceAfter=No\n",
    "43\t.\t.\tPUNCT\t.\t_\t10\tpunct\t_\t_\n",
    "\"\"\"\n",
    "\n",
    "keeping_example = \"\"\"\n",
    "1\tHe\the\tPRON\tPRP\tCase=Nom|Gender=Masc|Number=Sing|Person=3|PronType=Prs\t3\tnsubj\t_\t_\n",
    "2\talso\talso\tADV\tRB\t_\t3\tadvmod\t_\t_\n",
    "3\tordered\torder\tVERB\tVBD\tMood=Ind|Tense=Past|VerbForm=Fin\t0\troot\t_\t_\n",
    "4\tthat\tthat\tSCONJ\tIN\t_\t8\tmark\t_\t_\n",
    "5\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t6\tdet\t_\t_\n",
    "6\treceiver\treceiver\tNOUN\tNN\tNumber=Sing\t8\tnsubj\t_\t_\n",
    "7\tnot\tnot\tADV\tRB\t_\t8\tneg\t_\t_\n",
    "8\tget\tget\tVERB\tVB\tVerbForm=Inf\t3\tccomp\t_\t_\n",
    "9\ta\ta\tDET\tDT\tDefinite=Ind|PronType=Art\t15\tdet\t_\t_\n",
    "10\t(\t(\tPUNCT\t-LRB-\t_\t12\tpunct\t_\tSpaceAfter=No\n",
    "11\tI\tI\tPRON\tPRP\tCase=Nom|Number=Sing|Person=1|PronType=Prs\t12\tnsubj\t_\t_\n",
    "12\tthink\tthink\tVERB\tVBP\tMood=Ind|Tense=Pres|VerbForm=Fin\t15\tparataxis\t_\t_\n",
    "13\tchrome\tchrome\tNOUN\tNN\tNumber=Sing\t12\tdobj\t_\tSpaceAfter=No\n",
    "14\t)\t)\tPUNCT\t-RRB-\t_\t12\tpunct\t_\t_\n",
    "15\tcoating\tcoating\tNOUN\tNN\tNumber=Sing\t8\tdobj\t_\t_\n",
    "16\tthat\tthat\tDET\tWDT\tPronType=Rel\t17\tnsubj\t_\t_\n",
    "17\thelped\thelp\tVERB\tVBD\tMood=Ind|Tense=Past|VerbForm=Fin\t8\tacl:relcl\t_\t_\n",
    "18\tkeep\tkeep\tVERB\tVB\tVerbForm=Inf\t17\tccomp\t_\t_\n",
    "19\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t20\tdet\t_\t_\n",
    "20\tgun\tgun\tNOUN\tNN\tNumber=Sing\t18\tdobj\t_\t_\n",
    "21\tfrom\tfrom\tSCONJ\tIN\t_\t22\tmark\t_\t_\n",
    "22\tjamming\tjam\tVERB\tVBG\tVerbForm=Ger\t18\tadvcl\t_\tSpaceAfter=No\n",
    "23\t.\t.\tPUNCT\t.\t_\t3\tpunct\t_\t_\n",
    "\"\"\"\n",
    "\n",
    "benefit_example = \"\"\"\n",
    "1\tIt\tit\tPRON\tPRP\tCase=Nom|Gender=Neut|Number=Sing|Person=3|PronType=Prs\t2\texpl\t_\t_\n",
    "2\tsounds\tsound\tVERB\tVBZ\tMood=Ind|Number=Sing|Person=3|Tense=Pres|VerbForm=Fin\t0\troot\t_\t_\n",
    "3\tlike\tlike\tSCONJ\tIN\t_\t7\tmark\t_\t_\n",
    "4\tyour\tyou\tPRON\tPRP$\tPerson=2|Poss=Yes|PronType=Prs\t5\tnmod:poss\t_\t_\n",
    "5\tdog\tdog\tNOUN\tNN\tNumber=Sing\t7\tnsubj\t_\t_\n",
    "6\tmay\tmay\tAUX\tMD\tVerbForm=Fin\t7\taux\t_\t_\n",
    "7\tbenefit\tbenefit\tVERB\tVB\tVerbForm=Inf\t2\tadvcl\t_\t_\n",
    "8\tfrom\tfrom\tSCONJ\tIN\t_\t10\tmark\t_\t_\n",
    "9\tbeing\tbe\tAUX\tVBG\tVerbForm=Ger\t10\tauxpass\t_\t_\n",
    "10\thospitalized\thospitalize\tVERB\tVBN\tTense=Past|VerbForm=Part|Voice=Pass\t7\tadvcl\t_\tSpaceAfter=No\n",
    "11\t,\t,\tPUNCT\t,\t_\t10\tpunct\t_\t_\n",
    "12\tand\tand\tCONJ\tCC\t_\t10\tcc\t_\t_\n",
    "13\tput\tput\tVERB\tVBN\tTense=Past|VerbForm=Part\t10\tconj\t_\t_\n",
    "14\ton\ton\tADP\tIN\t_\t16\tcase\t_\t_\n",
    "15\tIV\tiv\tNOUN\tNN\tNumber=Sing\t16\tcompound\t_\t_\n",
    "16\tfluids\tfluid\tNOUN\tNNS\tNumber=Plur\t13\tnmod\t_\t_\n",
    "17\tto\tto\tPART\tTO\t_\t18\tmark\t_\t_\n",
    "18\thelp\thelp\tVERB\tVB\tVerbForm=Inf\t10\tadvcl\t_\t_\n",
    "19\this\the\tPRON\tPRP$\tGender=Masc|Number=Sing|Person=3|Poss=Yes|PronType=Prs\t20\tnmod:poss\t_\t_\n",
    "20\thydration\thydration\tNOUN\tNN\tNumber=Sing\t18\tdobj\t_\tSpaceAfter=No\n",
    "21\t,\t,\tPUNCT\t,\t_\t18\tpunct\t_\t_\n",
    "22\tadminister\tadminister\tVERB\tVB\tVerbForm=Inf\t18\tconj\t_\t_\n",
    "23\tthe\tthe\tDET\tDT\tDefinite=Def|PronType=Art\t25\tdet\t_\t_\n",
    "24\tantibiotics\tantibiotic\tNOUN\tNNS\tNumber=Plur\t25\tcompound\t_\t_\n",
    "25\tIV\tiv\tNOUN\tNN\tNumber=Sing\t22\tdobj\t_\tSpaceAfter=No\n",
    "26\t,\t,\tPUNCT\t,\t_\t18\tpunct\t_\t_\n",
    "27\tand\tand\tCONJ\tCC\t_\t18\tcc\t_\t_\n",
    "28\tto\tto\tPART\tTO\t_\t29\tmark\t_\t_\n",
    "29\tensure\tensure\tVERB\tVB\tVerbForm=Inf\t18\tconj\t_\t_\n",
    "30\tthat\tthat\tSCONJ\tIN\t_\t32\tmark\t_\t_\n",
    "31\the\the\tPRON\tPRP\tCase=Nom|Gender=Masc|Number=Sing|Person=3|PronType=Prs\t32\tnsubj\t_\t_\n",
    "32\tstarts\tstart\tVERB\tVBZ\tMood=Ind|Number=Sing|Person=3|Tense=Pres|VerbForm=Fin\t29\tccomp\t_\t_\n",
    "33\tto\tto\tPART\tTO\t_\t34\tmark\t_\t_\n",
    "34\trecover\trecover\tVERB\tVB\tVerbForm=Inf\t32\txcomp\t_\tSpaceAfter=No\n",
    "35\t.\t.\tPUNCT\t.\t_\t2\tpunct\t_\t_\n",
    "\"\"\"\n",
    "\n",
    "# examples = [prevent_example, delay_example, discourage_example, restrict_example, keeping_example]\n",
    "# examples = [delay_example, restrict_example]\n",
    "examples = [prevent_example, delay_example, discourage_example, keeping_example, benefit_example]\n",
    "            #, benefit_example]\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "for i, ex in enumerate(examples):\n",
    "    examples[i] = [ud_parse for sent_id, ud_parse in load_conllu(ex)][0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[Predicate(is/2), Predicate(prevented/6), Predicate(running/11)]\n",
      "One answer is that the Pentagon prevented the State Department from running the CPA .\n",
      "\t?a \u001b[32mis\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mOne answer\u001b[0m\n",
      "\t\t?b: \u001b[33mSOMETHING\u001b[0m := the Pentagon prevented the State Department from running the CPA\n",
      "\t?a \u001b[32mprevented\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mthe Pentagon\u001b[0m\n",
      "\t\t?b: \u001b[32mthe State Department\u001b[0m\n",
      "\t?a \u001b[32mrunning\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mthe State Department\u001b[0m\n",
      "\t\t?b: \u001b[32mthe CPA\u001b[0m\n",
      "[Predicate(agreed/3), Predicate(delay/5)]\n",
      "The judge had agreed to delay the civil rights matter from proceeding until at least October 2004 .\n",
      "\t?a \u001b[32mhad\u001b[0m \u001b[32magreed\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mThe judge\u001b[0m\n",
      "\t\t?b: \u001b[33mSOMETHING\u001b[0m := to delay the civil rights matter from proceeding until at least October 2004\n",
      "\t?a \u001b[32mdelay\u001b[0m ?b \u001b[32mfrom\u001b[0m ?c \u001b[32muntil\u001b[0m ?d\n",
      "\t\t?a: \u001b[32mThe judge\u001b[0m\n",
      "\t\t?b: \u001b[32mthe civil rights matter\u001b[0m\n",
      "\t\t?c: \u001b[32mproceeding\u001b[0m\n",
      "\t\t?d: \u001b[32mat least October 2004\u001b[0m\n",
      "[Predicate(found/9), Predicate(discouraged/17), Predicate(spending/21), Predicate(out/32), Predicate(munching/38)]\n",
      "Since reintroduction , however , Ripple and Beschta have found , fear of the wolves apparently has discouraged the elk from spending too much time at stream banks -- where they are out in the open -- and munching on the saplings .\n",
      "\t\u001b[32mSince\u001b[0m ?a \u001b[32m,\u001b[0m \u001b[32mhowever\u001b[0m \u001b[32m,\u001b[0m ?b \u001b[32mhave\u001b[0m \u001b[32mfound\u001b[0m ?c\n",
      "\t\t?a: \u001b[32mreintroduction\u001b[0m\n",
      "\t\t?b: \u001b[32mRipple and Beschta\u001b[0m\n",
      "\t\t?c: \u001b[33mSOMETHING\u001b[0m := fear of the wolves apparently has discouraged the elk from spending too much time at stream banks -- where they are out in the open -- and munching on the saplings\n",
      "\t?a \u001b[32mapparently\u001b[0m \u001b[32mhas\u001b[0m \u001b[32mdiscouraged\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mfear of the wolves\u001b[0m\n",
      "\t\t?b: \u001b[32mthe elk\u001b[0m\n",
      "\t?a \u001b[32mspending\u001b[0m ?b \u001b[32mat\u001b[0m ?c\n",
      "\t\t?a: \u001b[32mthe elk\u001b[0m\n",
      "\t\t?b: \u001b[32mtoo much time\u001b[0m\n",
      "\t\t?c: \u001b[32mstream banks -- where they are out in the open\u001b[0m\n",
      "\t?a \u001b[32mwhere\u001b[0m ?b \u001b[32mare\u001b[0m \u001b[32mout\u001b[0m \u001b[32min\u001b[0m ?c\n",
      "\t\t?a: \u001b[32mstream banks\u001b[0m\n",
      "\t\t?b: \u001b[32mthey\u001b[0m\n",
      "\t\t?c: \u001b[32mthe open\u001b[0m\n",
      "\t?a \u001b[32mmunching\u001b[0m \u001b[32mon\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mfear of the wolves\u001b[0m\n",
      "\t\t?b: \u001b[32mthe saplings\u001b[0m\n",
      "[Predicate(ordered/2), Predicate(get/7), Predicate(think/11), Predicate(helped/16), Predicate(keep/17), Predicate(jamming/21)]\n",
      "He also ordered that the receiver not get a ( I think chrome ) coating that helped keep the gun from jamming .\n",
      "\t?a \u001b[32malso\u001b[0m \u001b[32mordered\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mHe\u001b[0m\n",
      "\t\t?b: \u001b[33mSOMETHING\u001b[0m := the receiver not get a ( I think chrome ) coating that helped keep the gun from jamming\n",
      "\t?a \u001b[32mnot\u001b[0m \u001b[32mget\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mthe receiver\u001b[0m\n",
      "\t\t?b: \u001b[32ma ( I think chrome ) coating\u001b[0m\n",
      "\t?a \u001b[32mthink\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mI\u001b[0m\n",
      "\t\t?b: \u001b[32mchrome\u001b[0m\n",
      "\t?a \u001b[32mhelped\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mget\u001b[0m\n",
      "\t\t?b: \u001b[33mSOMETHING\u001b[0m := keep the gun from jamming\n",
      "\t?a \u001b[32mkeep\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mthat\u001b[0m\n",
      "\t\t?b: \u001b[32mthe gun\u001b[0m\n",
      "\t?a \u001b[32mjamming\u001b[0m\n",
      "\t\t?a: \u001b[32mthe gun\u001b[0m\n",
      "[Predicate(benefit/6), Predicate(hospitalized/9), Predicate(put/12), Predicate(help/17), Predicate(administer/21), Predicate(ensure/28), Predicate(starts/31), Predicate(recover/33)]\n",
      "It sounds like your dog may benefit from being hospitalized , and put on IV fluids to help his hydration , administer the antibiotics IV , and to ensure that he starts to recover .\n",
      "\t?a \u001b[32mmay\u001b[0m \u001b[32mbenefit\u001b[0m\n",
      "\t\t?a: \u001b[32myour dog\u001b[0m\n",
      "\t?a \u001b[32mbeing\u001b[0m \u001b[32mhospitalized\u001b[0m\n",
      "\t\t?a: \u001b[32myour dog\u001b[0m\n",
      "\t?a \u001b[32mput\u001b[0m \u001b[32mon\u001b[0m ?b\n",
      "\t\t?a: \u001b[32myour dog\u001b[0m\n",
      "\t\t?b: \u001b[32mIV fluids\u001b[0m\n",
      "\t?a \u001b[32mhelp\u001b[0m ?b\n",
      "\t\t?a: \u001b[32myour dog\u001b[0m\n",
      "\t\t?b: \u001b[32mhis hydration\u001b[0m\n",
      "\t?a \u001b[32madminister\u001b[0m ?b\n",
      "\t\t?a: \u001b[32myour dog\u001b[0m\n",
      "\t\t?b: \u001b[32mthe antibiotics IV\u001b[0m\n",
      "\t?a \u001b[32mensure\u001b[0m ?b\n",
      "\t\t?a: \u001b[32myour dog\u001b[0m\n",
      "\t\t?b: \u001b[33mSOMETHING\u001b[0m := he starts to recover\n",
      "\t?a \u001b[32mstarts\u001b[0m ?b\n",
      "\t\t?a: \u001b[32mhe\u001b[0m\n",
      "\t\t?b: \u001b[33mSOMETHING\u001b[0m := to recover\n",
      "\t?a \u001b[32mrecover\u001b[0m\n",
      "\t\t?a: \u001b[32mhe\u001b[0m\n"
     ]
    }
   ],
   "source": [
    "resolve_relcl = True  # relative clauses\n",
    "resolve_conj = False   # conjuction\n",
    "borrow_arg_for_relcl = True\n",
    "\n",
    "\n",
    "opts = PredPattOpts(\n",
    "        resolve_relcl=resolve_relcl,\n",
    "        borrow_arg_for_relcl=borrow_arg_for_relcl,\n",
    "        resolve_conj=resolve_conj,\n",
    "        cut=True\n",
    "        \n",
    "        )\n",
    "\n",
    "\n",
    "for ex in examples:\n",
    "    ppatt = PredPatt(ex, opts = opts)\n",
    "    print(ppatt.instances)\n",
    "    print(\" \".join([token.text for token in ppatt.tokens]))\n",
    "    print(ppatt.pprint(color=True))\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "PredPatt provides a straightforward way to show the predicate-argument content. You can enable/disable \"`color`\" and \"`track_rule`\" to generate the output that suits your needs."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "collapsed": true
   },
   "source": [
    "In practice, iterating the extracted predicates is quite easy. Let's iteratively print out the predicate and its arguments."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(\" \".join([token.text for token in ppatt.tokens]))\n",
    "for predicate in ppatt.instances:\n",
    "    print(\"\\t%s [%s-%s-%d]\" % (\" \".join(token.text for token in predicate.tokens),\n",
    "                            predicate.root.text, predicate.root.gov_rel, predicate.root.position))\n",
    "    for argument in predicate.arguments:\n",
    "        print(\"\\t\\t%s [%s-%s-%d]\" % (\" \".join(token.text for token in argument.tokens),\n",
    "                                     argument.root.text, argument.root.gov_rel, argument.root.position))"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### Configurable options\n",
    "PredPatt can be configured to tackle various syntactic phenomena. To do so, you can pass the configurable options to PredPatt."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Pierre Vinken , 61 years old , will join the board as a nonexecutive director Nov. 29 .\n",
      "\t?a \u001b[33mis/are\u001b[0m \u001b[32m61\u001b[0m \u001b[32myears\u001b[0m \u001b[32mold\u001b[0m\u001b[35m\t[old-amod]\u001b[0m\n",
      "\t\t?a: \u001b[32mPierre Vinken\u001b[0m\u001b[35m\t[Vinken-nsubj]\u001b[0m\n",
      "\t?a \u001b[32mwill\u001b[0m \u001b[32mjoin\u001b[0m ?b \u001b[32mas\u001b[0m ?c ?d\u001b[35m\t[join-root]\u001b[0m\n",
      "\t\t?a: \u001b[32mPierre Vinken , 61 years old\u001b[0m\u001b[35m\t[Vinken-nsubj]\u001b[0m\n",
      "\t\t?b: \u001b[32mthe board\u001b[0m\u001b[35m\t[board-dobj]\u001b[0m\n",
      "\t\t?c: \u001b[32ma nonexecutive director\u001b[0m\u001b[35m\t[director-nmod]\u001b[0m\n",
      "\t\t?d: \u001b[32mNov. 29\u001b[0m\u001b[35m\t[Nov.-nmod:tmod]\u001b[0m\n",
      "\t?a \u001b[33mis/are\u001b[0m \u001b[32mnonexecutive\u001b[0m\u001b[35m\t[nonexecutive-amod]\u001b[0m\n",
      "\t\t?a: \u001b[32ma director\u001b[0m\u001b[35m\t[director-nmod]\u001b[0m\n"
     ]
    }
   ],
   "source": [
    "from predpatt import PredPattOpts\n",
    "from predpatt.util.ud import dep_v1, dep_v2\n",
    "resolve_relcl = True  # relative clauses\n",
    "resolve_appos = True  # appositional modifiers\n",
    "resolve_amod = True   # adjectival modifiers\n",
    "resolve_conj = True   # conjuction\n",
    "resolve_poss = True   # possessives\n",
    "ud = dep_v1.VERSION   # the version of UD\n",
    "opts = PredPattOpts(\n",
    "        resolve_relcl=resolve_relcl,\n",
    "        resolve_appos=resolve_appos,\n",
    "        resolve_amod=resolve_amod,\n",
    "        resolve_conj=resolve_conj,\n",
    "        resolve_poss=resolve_poss,\n",
    "        ud=ud\n",
    "        )\n",
    "\n",
    "ppatt = PredPatt(concrete_example, opts=opts)\n",
    "print \" \".join([token.text for token in ppatt.tokens])\n",
    "print ppatt.pprint(color=True)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}