Fix typo in examples/recurrent_neural_network.py

masscollaborationlabs · Dec 20, 2017 · 9e8a5ca · 9e8a5ca
1 parent c5256ff
commit 9e8a5ca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mlfromscratch/deep_learning/layers.py b/mlfromscratch/deep_learning/layers.py
@@ -155,7 +155,7 @@ def forward_pass(self, X, training=True):
         for t in range(timesteps):
             # Input to state_t is the current input and output of previous states
             self.state_input[:, t] = X[:, t].dot(self.U.T) + self.states[:, t-1].dot(self.W.T)
-            self.states[:, t] = self.activation.function(self.state_input[:, t])
+            self.states[:, t] = self.activation(self.state_input[:, t])
             self.outputs[:, t] = self.states[:, t].dot(self.V.T)
 
         return self.outputs

diff --git a/mlfromscratch/examples/recurrent_neural_network.py b/mlfromscratch/examples/recurrent_neural_network.py
@@ -5,7 +5,7 @@
 from mlfromscratch.deep_learning import NeuralNetwork
 from mlfromscratch.utils import train_test_split, to_categorical, normalize, Plot
 from mlfromscratch.utils import get_random_subsets, shuffle_data, accuracy_score
-from mlfromscratch.deep_learning.optimizers import GradientDescent, Adam, RMSprop, Adagrad, Adadelta
+from mlfromscratch.deep_learning.optimizers import StochasticGradientDescent, Adam, RMSprop, Adagrad, Adadelta
 from mlfromscratch.deep_learning.loss_functions import CrossEntropy
 from mlfromscratch.utils.misc import bar_widgets
 from mlfromscratch.deep_learning.layers import RNN, Activation