arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Replace n_inputs with n_outputs, fixes #125

Browse Source
This commit is contained in:
Aurélien Geron
2017-12-07 18:57:30 -08:00
parent 53c2133bc7
commit 63c1523528
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
04_training_linear_models.ipynb
View File

@@ -1909,7 +1909,7 @@
" error = Y_proba - Y_train_one_hot\n", " error = Y_proba - Y_train_one_hot\n",
" if iteration % 500 == 0:\n", " if iteration % 500 == 0:\n",
" print(iteration, loss)\n", " print(iteration, loss)\n",
" gradients = 1/m * X_train.T.dot(error) + np.r_[np.zeros([1, n_inputs]), alpha * Theta[1:]]\n", " gradients = 1/m * X_train.T.dot(error) + np.r_[np.zeros([1, n_outputs]), alpha * Theta[1:]]\n",
" Theta = Theta - eta * gradients" " Theta = Theta - eta * gradients"
] ]
}, },
@@ -1987,7 +1987,7 @@
" l2_loss = 1/2 * np.sum(np.square(Theta[1:]))\n", " l2_loss = 1/2 * np.sum(np.square(Theta[1:]))\n",
" loss = xentropy_loss + alpha * l2_loss\n", " loss = xentropy_loss + alpha * l2_loss\n",
" error = Y_proba - Y_train_one_hot\n", " error = Y_proba - Y_train_one_hot\n",
" gradients = 1/m * X_train.T.dot(error) + np.r_[np.zeros([1, n_inputs]), alpha * Theta[1:]]\n", " gradients = 1/m * X_train.T.dot(error) + np.r_[np.zeros([1, n_outputs]), alpha * Theta[1:]]\n",
" Theta = Theta - eta * gradients\n", " Theta = Theta - eta * gradients\n",
"\n", "\n",
" logits = X_valid.dot(Theta)\n", " logits = X_valid.dot(Theta)\n",