From 5e38505cd7f5611becedef98d068a2f43f69f171 Mon Sep 17 00:00:00 2001
From: Donne Martin <donne.martin@gmail.com>
Date: Mon, 23 Mar 2015 07:24:12 -0400
Subject: [PATCH] Added random forest classification report.

---
 kaggle/titanic.ipynb | 57 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 56 insertions(+), 1 deletion(-)

diff --git a/kaggle/titanic.ipynb b/kaggle/titanic.ipynb
index f7f9037..2e3402a 100644
--- a/kaggle/titanic.ipynb
+++ b/kaggle/titanic.ipynb
@@ -1,7 +1,7 @@
 {
  "metadata": {
   "name": "",
-  "signature": "sha256:813d2b8b5a05ecbfafc2ce418b6370897bdd4d693bd80accac706955616ce774"
+  "signature": "sha256:fc5c7ecfd36fae36306417f49f9512178629710cef9c48066a4bddbefd059be4"
  },
  "nbformat": 3,
  "nbformat_minor": 0,
@@ -2959,6 +2959,61 @@
       }
      ],
      "prompt_number": 46
+    },
+    {
+     "cell_type": "markdown",
+     "metadata": {},
+     "source": [
+      "Display the classification report:"
+     ]
+    },
+    {
+     "cell_type": "markdown",
+     "metadata": {},
+     "source": [
+      "$$Precision = \\frac{TP}{TP + FP}$$ \n",
+      "\n",
+      "$$Recall = \\frac{TP}{TP + FN}$$ \n",
+      "\n",
+      "$$F1 = \\frac{2TP}{2TP + FP + FN}$$ "
+     ]
+    },
+    {
+     "cell_type": "code",
+     "collapsed": false,
+     "input": [
+      "from sklearn.metrics import classification_report\n",
+      "print(classification_report(test_y, \n",
+      "                            predict_y, \n",
+      "                            target_names=['Not Survived', 'Survived']))"
+     ],
+     "language": "python",
+     "metadata": {},
+     "outputs": [
+      {
+       "output_type": "stream",
+       "stream": "stdout",
+       "text": [
+        "              precision    recall  f1-score   support\n",
+        "\n",
+        "Not Survived       0.85      0.91      0.88       110\n",
+        "    Survived       0.84      0.74      0.78        69\n",
+        "\n",
+        " avg / total       0.84      0.84      0.84       179\n",
+        "\n"
+       ]
+      }
+     ],
+     "prompt_number": 47
+    },
+    {
+     "cell_type": "code",
+     "collapsed": false,
+     "input": [],
+     "language": "python",
+     "metadata": {},
+     "outputs": [],
+     "prompt_number": 47
     }
    ],
    "metadata": {}