Added activation function and increased epochs

Added non-linear ReLU activation function in order to make the network
a non-linear function approximator which is usually a wanted property
when using neural networks.
The increase in epochs is necessary to see a significant difference
between using an activation function and not using one.

Author: juelg

site/en/tutorials/load_data/csv.ipynb

@@ -226,7 +226,7 @@
       "outputs": [],
       "source": [
         "abalone_model = tf.keras.Sequential([\n",
-        "  layers.Dense(64),\n",
+        "  layers.Dense(64, activation='relu'),\n",
         "  layers.Dense(1)\n",
         "])\n",
         "\n",
@@ -251,7 +251,7 @@
       },
       "outputs": [],
       "source": [
-        "abalone_model.fit(abalone_features, abalone_labels, epochs=10)"
+        "abalone_model.fit(abalone_features, abalone_labels, epochs=30)"
       ]
     },
     {
@@ -337,14 +337,14 @@
       "source": [
         "norm_abalone_model = tf.keras.Sequential([\n",
         "  normalize,\n",
-        "  layers.Dense(64),\n",
+        "  layers.Dense(64, activation='relu'),\n",
         "  layers.Dense(1)\n",
         "])\n",
         "\n",
         "norm_abalone_model.compile(loss = tf.keras.losses.MeanSquaredError(),\n",
         "                           optimizer = tf.keras.optimizers.Adam())\n",
         "\n",
-        "norm_abalone_model.fit(abalone_features, abalone_labels, epochs=10)"
+        "norm_abalone_model.fit(abalone_features, abalone_labels, epochs=30)"
       ]
     },
     {
@@ -646,7 +646,7 @@
       "source": [
         "def titanic_model(preprocessing_head, inputs):\n",
         "  body = tf.keras.Sequential([\n",
-        "    layers.Dense(64),\n",
+        "    layers.Dense(64, activation='relu'),\n",
         "    layers.Dense(1)\n",
         "  ])\n",
         "\n",
@@ -678,7 +678,7 @@
       },
       "outputs": [],
       "source": [
-        "titanic_model.fit(x=titanic_features_dict, y=titanic_labels, epochs=10)"
+        "titanic_model.fit(x=titanic_features_dict, y=titanic_labels, epochs=30)"
       ]
     },
     {