diff --git a/datasets/Titanic.ipynb b/datasets/Titanic.ipynb
index 99d4865..f79912e 100644
--- a/datasets/Titanic.ipynb
+++ b/datasets/Titanic.ipynb
@@ -10,7 +10,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "id": "ecd595e6-4de0-45f1-8022-92ad3c47e540",
    "metadata": {},
    "outputs": [
@@ -32,7 +32,14 @@
     "import pandas as pd\n",
     "import numpy as np\n",
     "\n",
-    "df = pd.read_csv('titanic.csv')"
+    "df = pd.read_csv('titanic.csv')\n",
+    "\n",
+    "# Replicate TensorFlow's specific feature engineering schema\n",
+    "df = df.rename(columns={\n",
+    "    'Survived': 'survived', 'Sex': 'sex', 'Age': 'age',\n",
+    "    'SibSp': 'n_siblings_spouses', 'Parch': 'parch', 'Fare': 'fare',\n",
+    "    'Pclass': 'class', 'Embarked': 'embark_town'\n",
+    "})"
    ]
   },
   {