add descriptions to the onehot_ function

jasgnft · Apr 21, 2020 · f9df32f · f9df32f
1 parent 5d63c16
commit f9df32f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/ch12/ch12.ipynb b/ch12/ch12.ipynb
@@ -696,6 +696,8 @@
     "        ------------\n",
     "        y : array, shape = [n_examples]\n",
     "            Target values.\n",
+    "        n_classes : int\n",
+    "            Number of classes\n",
     "\n",
     "        Returns\n",
     "        -----------\n",
@@ -1339,4 +1341,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}
diff --git a/ch12/ch12.py b/ch12/ch12.py
@@ -308,6 +308,8 @@ def _onehot(self, y, n_classes):
         ------------
         y : array, shape = [n_examples]
             Target values.
+        n_classes : int
+            Number of classes
 
         Returns
         -----------

diff --git a/ch12/neuralnet.py b/ch12/neuralnet.py
@@ -49,6 +49,8 @@ def _onehot(self, y, n_classes):
         ------------
         y : array, shape = [n_examples]
             Target values.
+        n_classes : int
+            Number of classes
 
         Returns
         -----------