diff --git a/tf_2.x/lab-06-1-softmax_classifier-eager.ipynb b/tf_2.x/lab-06-1-softmax_classifier-eager.ipynb index 08ff21e..2308602 100644 --- a/tf_2.x/lab-06-1-softmax_classifier-eager.ipynb +++ b/tf_2.x/lab-06-1-softmax_classifier-eager.ipynb @@ -368,9 +368,9 @@ } ], "source": [ - "class softmax_classifer(tf.keras.Model):\n", + "class softmax_classifier(tf.keras.Model):\n", " def __init__(self, nb_classes):\n", - " super(softmax_classifer, self).__init__()\n", + " super(softmax_classifier, self).__init__()\n", " self.W = tf.Variable(tf.random.normal((4, nb_classes)), name='weight')\n", " self.b = tf.Variable(tf.random.normal((nb_classes,)), name='bias')\n", " \n", @@ -397,7 +397,7 @@ " if (i==0) | ((i+1)%verbose==0):\n", " print('Loss at epoch %d: %f' %(i+1, self.cost_fn(X, Y).numpy()))\n", " \n", - "model = softmax_classifer(nb_classes)\n", + "model = softmax_classifier(nb_classes)\n", "model.fit(x_data, y_data)" ] },