Remove unnecessary tree model hidden_dim args (h/t @zijwang)

shishirpy · Apr 8, 2019 · e29d8aa · e29d8aa
1 parent c264cc4
commit e29d8aa
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/sst_03_neural_networks.ipynb b/sst_03_neural_networks.ipynb
@@ -972,7 +972,6 @@
     "    sst_glove_vocab,\n",
     "    embedding=glove_embedding,\n",
     "    embed_dim=None,  # Ignored when embedding is not `None`\n",
-    "    hidden_dim=50,\n",
     "    max_iter=10,\n",
     "    eta=0.05) "
    ]
@@ -1062,7 +1061,6 @@
     "torch_tree_nn_glove = TorchTreeNN(\n",
     "    sst_glove_vocab,\n",
     "    embedding=glove_embedding,\n",
-    "    hidden_dim=50,\n",
     "    embed_dim=50,\n",
     "    max_iter=10,\n",
     "    eta=0.05)"
@@ -1160,7 +1158,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.1"
+   "version": "3.7.3"
   },
   "widgets": {
    "state": {},