Adding dense layer as a child block

deepjavalibrary · Mar 10, 2021 · 7591ef4 · 7591ef4
1 parent 122e8bf
commit 7591ef4
Showing 1 changed file with 1 addition and 22 deletions.
diff --git a/chapter_recurrent-neural-networks/rnn-concise.ipynb b/chapter_recurrent-neural-networks/rnn-concise.ipynb
@@ -50,28 +50,6 @@
     "%load ../utils/Accumulator.java\n",
     "%load ../utils/Animator.java\n",
     "%load ../utils/Training.java\n",
-    "\n",
-    "\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/Functions.java\n",
-    "// // %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/PlotUtils.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/StopWatch.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/Accumulator.java\n",
-    "// // %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/Animator.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/Training.java\n",
-    "\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/RNNModelScratch.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/Vocab.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/SeqDataLoader.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/RNNModel.java\n",
-    "// %load /Users/nivesmn/Documents/projects/intellij_notebooks/section8_6/src/main/java/TimeMachine.java"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 3,
-   "metadata": {},
-   "outputs": [],
-   "source": [
     "%load ../utils/TimeMachineUtils.java"
    ]
   },
@@ -295,6 +273,7 @@
     "        this.addChildBlock(\"rnn\", rnnLayer);\n",
     "        this.vocabSize = vocabSize;\n",
     "        this.dense = Linear.builder().setUnits(vocabSize).build();\n",
+    "        this.addChildBlock(\"linear\", dense);\n",
     "    }\n",
     "\n",
     "    \n",