docstrings

ghellstern · Mar 22, 2017 · c324229 · c324229
1 parent 4a5dacc
commit c324229
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/week6.5/seq2seq.ipynb b/week6.5/seq2seq.ipynb
@@ -8,17 +8,12 @@
    },
    "outputs": [],
    "source": [
-    "%env THEANO_FLAGS=device=gpu7\n",
     "import numpy as np\n",
     "import theano\n",
     "import theano.tensor as T\n",
     "import lasagne\n",
     "import os\n",
-    "import pandas as pd\n",
-    "from tqdm import tqdm_notebook\n",
-    "\n",
-    "#thanks Muammar \n",
-    "PAD_ix=-1"
+    "import pandas as pd"
    ]
   },
   {
@@ -120,7 +115,11 @@
    },
    "outputs": [],
    "source": [
-    "def as_matrix(sequences,token_to_i, max_len=None,PAX_ix=PAD_ix):\n",
+    "def as_matrix(sequences,token_to_i, max_len=None,PAX_ix=-1):\n",
+    "    \"\"\"\n",
+    "    Converts several sequences of tokens to a matrix, edible a neural network.\n",
+    "    Crops at max_len(if given), pads shorter sequences with -1 or PAD_ix.\n",
+    "    \"\"\"\n",
     "    max_len = max_len or max(map(len,sequences))\n",
     "    \n",
     "    matrix = np.zeros((len(sequences),max_len),dtype='int8') -1\n",