Update Attention Visualization.ipynb

yufengm · Nov 17, 2017 · 406a1f2 · 406a1f2
1 parent 4756636
commit 406a1f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Attention Visualization.ipynb b/Attention Visualization.ipynb
@@ -350,8 +350,8 @@
     "    # print weights\n",
     "\n",
     "    vector = weights.sum( 0 )\n",
-    "    vector = vector / vector.sum( 1 )[ 0,0 ]\n",
-    "    att, ids_to_show =  vector.sort( 1, descending=True )\n",
+    "    vector = vector / vector.sum()\n",
+    "    att, ids_to_show =  vector.sort( 0, descending=True )\n",
     "\n",
     "    ids_to_show = ids_to_show.squeeze( 0 ).tolist()[:words2show]\n",
     "\n",